Update app.py

app.py

@@ -72,26 +72,28 @@ if uploaded:
         "Story:\n"
     )
 
-    # Generate Story
+    # Generate Story with full payload dict
+    payload = {
+        "inputs": story_prompt,
+        "parameters": {
+            "max_new_tokens":       200,
+            "temperature":          0.8,
+            "top_p":                0.95,
+            "repetition_penalty":   1.15,
+            "do_sample":            True,
+            "no_repeat_ngram_size": 2
+        }
+    }
+
     with st.spinner("📝 Writing magical story..."):
         try:
-            story_response = story_client(
-                story_prompt,
-                parameters={
-                    "max_new_tokens":       200,
-                    "temperature":          0.8,
-                    "top_p":                0.95,
-                    "repetition_penalty":   1.15,
-                    "do_sample":            True,
-                    "no_repeat_ngram_size": 2
-                }
-            )
-            full_text = story_response[0].get("generated_text", "")
-            story = full_text.split("Story:")[-1].strip()
+            story_response = story_client(payload)
+            full_text      = story_response[0].get("generated_text", "")
+            story          = full_text.split("Story:")[-1].strip()
             if "." in story:
                 story = story.rsplit(".", 1)[0] + "."
         except Exception as e:
-            st.error(f"🚨 Story magic failed: {str(e)}")
+            st.error(f"🚨 Story magic failed: {e}")
             st.stop()
 
     # Display Story
@@ -111,3 +113,4 @@ if uploaded:
 # Footer
 st.markdown("---\n*Made with ❤️ by your friendly story wizard*")
 
+