Update app.py

app.py

@@ -15,26 +15,10 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    pipe = pipeline(
-        "text-generation", 
-        model="pranavpsv/genre-story-generator-v2",
-        max_new_tokens=300,            # 预留冗余空间
-        min_new_tokens=100,
-        do_sample=True,
-        temperature=0.65,             # 降低随机性
-        top_p=0.85,
-        top_k=30,                     # 加速生成
-        repetition_penalty=1.3,
-        no_repeat_ngram_size=3,       # 防止重复短语
-        early_stopping=False          # 避免提前终止
-    )
-    
-    # 生成并后处理
-    story = pipe(text, return_full_text=False)[0]['generated_text']
+    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_text = pipe(text)[0]['generated_text']
+    return story_text
 
-    # 智能截断机制
-    target_length = min(max(len(story.split()), 50), 150)
-    return " ".join(story.split()[:target_length])
 
 # text2audio
 def text2audio(story_text):