Update app.py

app.py

@@ -14,6 +14,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
+    # Use DistilGPT-2 for text generation
     story_generator = pipeline("text-generation", model="distilgpt2")
     # Generate a story with a maximum of 90 words
     story = story_generator(text, max_length=90, num_return_sequences=1)[0]["generated_text"]
@@ -25,7 +26,8 @@ def text2story(text):
 
 # text2audio
 def text2audio(story_text):
-    tts_pipeline = pipeline("text-to-speech", model="espnet/kan-bayashi_ljspeech_vits")
+    # Use FastSpeech 2 for text-to-speech
+    tts_pipeline = pipeline("text-to-speech", model="facebook/fastspeech2-en-ljspeech")
     audio_data = tts_pipeline(story_text)
     return audio_data