Update app.py

app.py

@@ -73,40 +73,35 @@ else:
         "Story:"
     )
 
-    # 4) Generate story
-    with st.spinner("✍️ Generating story..."):
-        try:
-            story_out = story_client(
-                inputs=prompt,
-                parameters={
-                    "max_new_tokens": 250,
-                    "temperature": 0.7,
-                    "top_p": 0.9,
-                    "repetition_penalty": 1.1,
-                    "do_sample": True
-                }
-            )
-            
-            # Parse response
-            if isinstance(story_out, list):
-                story_text = story_out[0].get("generated_text", "")
-            else:
-                story_text = story_out.get("generated_text", "")
-            
-            # Clean up output
-            story = story_text.split("Story:")[-1].strip()
-
-        except Exception as e:
-            st.error(f"🚨 Story generation failed: {str(e)}")
-            st.stop()
 
-    if not story:
-        st.error("😕 Couldn’t generate a story. Please try again!")
+复制代码
+# 4) Generate story with corrected parameter format
+with st.spinner("✍️ Generating story..."):
+    try:
+        story_out = story_client(
+            prompt,
+            max_new_tokens=250,  # Direct keyword arguments
+            temperature=0.7,
+            top_p=0.9,
+            top_k=50,
+            repetition_penalty=1.1,
+            do_sample=True,
+            no_repeat_ngram_size=2
+        )
+
+        # Handle response format
+        if isinstance(story_out, list):
+            story_text = story_out[0].get("generated_text", "")
+        else:  # Handle single-dictionary response
+            story_text = story_out.get("generated_text", "")
+        
+        # Extract story content after last prompt mention
+        story = story_text.split("Story:")[-1].strip()
+
+    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: