Update app.py

app.py

@@ -31,11 +31,12 @@ def translate_speech(audio):
     translated_text = translator(transcription, return_tensors="pt")
     print(f"Translated text: {translated_text}")  # Print the translated text to see what it contains
 
-    # Check if the translated text contains 'generated_text'
-    if 'generated_text' in translated_text[0]:
-        translated_text_str = translated_text[0]['generated_text']
+    # Check if the translated text contains 'generated_token_ids'
+    if 'generated_token_ids' in translated_text[0]:
+        # Decode the tokens into text
+        translated_text_str = translator.tokenizer.decode(translated_text[0]['generated_token_ids'])
     else:
-        print("The translated text does not contain 'generated_text'")
+        print("The translated text does not contain 'generated_token_ids'")
         return
 
     # Use the text-to-speech pipeline to synthesize the translated text