Update app.py

app.py

@@ -42,19 +42,19 @@ if uploaded:
 
     # —––––––– 2. Story generation
     prompt = (
-        "Write a playful, 50–100 word story for 3–10 year-old children "
+        "Write a 80–100 words story for kids"
         f"based on this description:\n\n“{cap}”\n\nStory:"
     )
     with st.spinner("✍️ Writing a story..."):
-        out = storyteller(
-            prompt,
-            max_length=200,
-            do_sample=True,
-            top_p=0.9,
-            temperature=0.8,
-            num_return_sequences=1
-        )
-        story = out[0]["generated_text"].strip()
+    out = storyteller(
+        prompt,
+        max_new_tokens=120, 
+        do_sample=True,
+        top_p=0.9,
+        temperature=0.8,
+        num_return_sequences=1
+)
+    story = out[0]["generated_text"].strip()
     st.markdown("**Story:**")
     st.write(story)