Update app.py

app.py

@@ -68,39 +68,36 @@ else:
         "Story:"
     )
 
-    # 4) Generate story with CORRECT parameters
-    with st.spinner("✍️ Generating story..."):
-        try:
-            story_out = story_client(
-                inputs=prompt,
-                max_new_tokens=150,
-                do_sample=True,
-                temperature=0.8,
-                top_p=0.9,
-                repetition_penalty=1.1,
-                no_repeat_ngram_size=2
-            )
-            
-            # Handle different response formats
-            if isinstance(story_out, list):
-                story_text = story_out[0].get("generated_text", "")
-            else:
-                story_text = story_out.get("generated_text", "")
-            
-            # Clean up the output
-            story = story_text.split("Story:")[-1].strip()
-
-        except Exception as e:
-            st.error(f"🚨 Story generation failed: {str(e)}")
-            st.stop()
+   # 4) Generate story with proper parameter structure
+with st.spinner("✍️ Generating story..."):
+    try:
+        # Generation parameters must be nested under 'parameters'
+        story_out = story_client(
+            inputs=prompt,
+            parameters={
+                "max_new_tokens": 150,
+                "do_sample": True,
+                "temperature": 0.8,
+                "top_p": 0.9,
+                "top_k": 50,
+                "repetition_penalty": 1.1,
+                "no_repeat_ngram_size": 2
+            }
+        )
+
+        # Handle different response formats
+        if isinstance(story_out, list):
+            story_text = story_out[0].get("generated_text", "")
+        else:
+            story_text = story_out.get("generated_text", "")
+        
+        # Extract story content after the prompt
+        story = story_text.split("Story:")[-1].strip()
 
-    if not story:
-        st.error("😕 Couldn’t generate a story. Please try again!")
+    except Exception as e:
+        st.error(f"🚨 Story generation failed: {str(e)}")
         st.stop()
 
-    st.markdown("**Story:**")
-    st.write(story)
-
     # 5) Text-to-Speech
     with st.spinner("🔊 Converting to speech..."):
         try: