Update app.py

app.py

@@ -13,19 +13,16 @@ def img2text(url):
 def text2story(text):
     pipe = pipeline("text-generation", 
                     model="pranavpsv/genre-story-generator-v2",
-                    max_new_tokens=100,  # 控制生成内容长度
-                    truncation=True)      # 确保输入不被截断
+                    max_new_tokens=100,
+                    truncation=True)     
     story_text = pipe(text, 
-                      do_sample=True,     # 启用随机采样
-                      temperature=0.9,    # 控制随机性（0-1，越大越有创意）
-                      top_k=50,           # 限制采样词汇量
+                      do_sample=True,     
+                      temperature=0.9,    
+                      top_k=50,           
                       num_return_sequences=1)[0]['generated_text']
-    
-    # 确保输出以完整句子结尾
     last_punctuation = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
     if last_punctuation != -1:
         story_text = story_text[:last_punctuation+1]
-        
     safe_story = story_text.replace("violence", "").replace("scary", "").replace("died", "solved")
     return safe_story