Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from transformers import pipeline
 from gtts import gTTS  # Google Text-to-Speech library
-import IPython.display as ipd  # For playing audio in the notebook
 
 # Create a translation pipeline
 pipe = pipeline('translation', model='Helsinki-NLP/opus-mt-en-hi')
@@ -11,15 +10,12 @@ if text:
     out = pipe(text, src_lang='en', tgt_lang='hi')
     st.json(out)
     
-    # Extract the translated text from the JSON output
-    translation_text = out[0]['translation_text']
-    
     # Convert the translated text to speech
+    translation_text = out[0]['translation_text']
     tts = gTTS(translation_text, lang='hi')  # You can specify the desired language ('hi' for Hindi)
     
-    # Save the generated speech as an audio file (e.g., "translated_audio.mp3")
-    audio_path = "translated_audio.mp3"
-    tts.save(audio_path)
-    
-    # Display the audio player for listening to the speech
-    st.audio(audio_path, format='audio/mp3')
+    # Save the generated speech as an audio file
+    tts.save("translated_audio.mp3")
+
+    # Display the audio player
+    st.audio("translated_audio.mp3")