Update app.py

app.py

@@ -78,9 +78,9 @@ def predict(message, history):
         repetition_penalty=1.1,
     )
 
-    # 使用线程来运行生成过程
-    t = Thread(target=model.generate, kwargs=generate_kwargs)
-    t.start()
+    # # 使用线程来运行生成过程
+    # t = Thread(target=model.generate, kwargs=generate_kwargs)
+    # t.start()
 
     # 实时生成部分消息
     partial_message = ""
@@ -90,7 +90,8 @@ def predict(message, history):
             break
         # 将历史记录和当前消息转换为 tuple 格式并实时返回
         # yield [(msg, bot) for msg, bot in history] + [(message, partial_message)]
-        yield (message, partial_message)
+        # yield (message, partial_message)
+        yield message
     
     # # 处理生成的最终回复
     # final_message = partial_message.replace(sft_end_token, "").strip()