Update app.py

app.py

@@ -20,13 +20,13 @@ def generate_caption(image_file):
 def generate_story(caption):
     story_generator = pipeline(
         "text-generation",
-        model="Qwen/Qwen2-1.5B",
+        model="Qwen/Qwen2.5-0.5B-Instruct",
     )
     prompt = (
-        "You are a talented children's story writer renowned for your creativity and captivating narratives. "
-        "Using the image details provided below, please compose an enchanting tale tailored for children aged 3 to 10. "
-        "Rather than simply rephrasing the image details, enrich your story with imaginative characters, quirky adventures, "
-        "and delightful surprises. Let your narrative flow naturally and spark wonder in your young audience. "
+        "You are a highly imaginative children's story writer celebrated for your creativity and captivating narratives. "
+        "Using the image details provided below, please craft an enchanting tale tailored for children aged 3 to 10. "
+        "Rather than simply reiterating the image details, enhance your story with imaginative characters, quirky adventures, "
+        "and delightful surprises that ignite wonder in every young heart. Let your narrative flow naturally and kindle the magic of storytelling. "
         "Please ensure that your story is engaging, coherent, and falls between 100 and 300 words in length.\n\n"
         f"Image Details: {caption}\n\nStory:"
     )