Update app.py

app.py

@@ -29,9 +29,12 @@ def generate_story(caption):
         device=device_id
     )
     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"
-    )
+                    "You are a creative children's story writer. Based on the following image details, "
+                    "please write an imaginative story for children aged 3-10. Do not simply rephrase the image details; "
+                    "instead, expand creatively by adding fun characters, adventures, and unexpected twists. "
+                    "The story must be at least 100 words long.\n\n"
+                    f"Image Details: {caption}\n\nStory:"
+                )
     result = story_generator(messages, max_length=300, num_return_sequences=1)
     story = result[0]['generated_text']
     return story