Update app.py

app.py

@@ -13,6 +13,8 @@ tokenizer = MarianTokenizer.from_pretrained(model_name)
 tts_model_name = "microsoft/speecht5_tts"
 tts_model = SpeechT5ForTextToSpeech.from_pretrained(tts_model_name)
 processor = SpeechT5Processor.from_pretrained(tts_model_name)
+
+# Load speaker embeddings
 speaker_embeddings = torch.tensor(load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")["xvector"][0]).unsqueeze(0)
 
 # Function to translate text
@@ -30,6 +32,17 @@ def synthesize_speech(text, target_lang):
     # Save the speech to a file
     output_path = "output.wav"
     sf.write(output_path, speech.numpy(), samplerate=16000)
+
+    # Check if the audio file was generated correctly
+    try:
+        with open(output_path, 'rb') as f:
+            audio_data = f.read()
+        if not audio_data:
+            st.error("Error: The audio file is empty.")
+        else:
+            st.success("Audio generated successfully.")
+    except Exception as e:
+        st.error(f"Error reading the audio file: {e}")
     
     return output_path