Update app.py

app.py

@@ -13,13 +13,8 @@ def generate_caption(image_file):
     return caption
 
 def generate_story(caption):
-    story_generator = pipeline("text-generation", model="Qwen/QwQ-32B")
-    messages = [
-        {
-            "role": "user",
-            "content": f"Please based on following image caption: '{caption}', generate a complete fairy tale story for children with at least 100 words and max 300 words"
-        }
-    ]
+    story_generator = pipeline("text-generation", model="Qwen/Qwen2-1.5B")
+    messages = "Please based on following image caption: "+caption+", generate a complete fairy tale story for children with at least 100 words and max 300 words"
     result = story_generator(messages, max_length=300, num_return_sequences=1)
     story = result[0]['generated_text']
     return story