Update app.py

app.py

@@ -31,7 +31,13 @@ def generate_story(caption):
         f"Image Details: {caption}\n\nStory:"
     )
     
-    result = story_generator(prompt, num_return_sequences=1)
+    result = story_generator(
+        prompt, 
+        max_length=300,
+        num_return_sequences=1,
+        do_sample=True,
+        temperature=1.0
+    )
     full_text = result[0]['generated_text']
     
     if "Story:" in full_text: