Update app.py

app.py

@@ -8,8 +8,8 @@ import os
 def load_models():
     return {
         "image_to_text": pipeline("image-to-text", model="Salesforce/blip-image-captioning-base"),
-        "story_gen": pipeline("text-generation", model="distilbert/distilgpt2"),
-        "text_to_speech": pipeline("text-to-speech", model="facebook/mms-tts-yor")
+        "story_gen": pipeline("text-generation", model="Qwen/Qwen2.5-1.5B-Instruct"),
+        "text_to_speech": pipeline("text-to-speech", model="facebook/mms-tts-eng")
     }
 
 # function part
@@ -18,7 +18,7 @@ def img2text(url, processor):
     return text
 
 def text2story(text, generator, max_length=150):
-    prompt = f"Create a children's story based on: {text}"
+    prompt = f"Create a story based on: {text}"
     story = generator(
         prompt,
         max_length=max_length,