Update app.py

app.py

@@ -3,7 +3,8 @@ from transformers import pipeline
 from gtts import gTTS 
 from PIL import Image
 import time
-
+from playsound import playsound
+ 
 @st.cache(allow_output_mutation=True)
 def load_model():
   """Retrieves the trained model"""
@@ -35,6 +36,7 @@ def main():
     st.write(generated_text)
     generated_audio = gTTS(generated_text)
     generated_audio.save('demo.mp3')
+    playsound('demo.mp3')
     audio_file = open('demo.mp3', 'rb')
     audio_bytes = audio_file.read()
     st.audio(audio_bytes, format='audio/ogg',start_time=0)