Update app.py

app.py

@@ -60,12 +60,12 @@ def generate_caption(image_bytes):
 # Function to generate a story from a caption
 def generate_story(caption):
     try:
-        prompt = f"Based on this text:'{caption}', ellaborate a short story with no more than 100 words for young children."
+        prompt = [f"Based on this text:'{caption}', ellaborate a short story with no more than 100 words for young children."]
         story_output = story_generator(
             prompt,
             max_length=150,
             do_sample=True,
-            temperature=0.7,
+            temperature=0.2,
             top_p=0.9,
             num_return_sequences=1
         )