Update app.py

app.py

@@ -13,8 +13,10 @@ def generate_caption(image_file):
     return caption
 
 
-def generate_story(prompt):
+def generate_story(caption):
+
     story_generator = pipeline("text-generation", model="gpt2")
+    prompt = f"Please based on following image caption: '{caption}', generate a complete fairy tale story for children with at least 100 words. "
     result = story_generator(prompt, max_length=300, num_return_sequences=1)
     story = result[0]['generated_text']