Update app.py

app.py

@@ -12,12 +12,12 @@ def img2text(url):
 # text2story
 def text2story(text):
     text_generation_model = pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
-    story_text = text_generation_model(text)[0]["generated_text"]
+    story_text = text_generation_model(text, min_length=100, max_length=150)[0]["generated_text"]
     return story_text
 
 # text2audio
 def text2audio(story_text):
-    text_to_speech_model = pipeline("text-to-speech", model="suno/bark-small")
+    text_to_speech_model = pipeline("text-to-speech", model="facebook/mms-tts-eng")
     audio_data = text_to_speech_model(story_text)
     return {
         'audio': audio_data["audio"],