Update app.py

app.py

@@ -127,29 +127,10 @@ def chat():
             temperature=temperature,
             top_p=top_p,
         )
+        model_output = model_output[len(prompt):].split(f"\n{USER_NAME}")[0]
         chat_history = chat_history + [[message, model_output]]
         yield chat_history
         return
-        # acc_text = ""
-        # for idx, response in enumerate(stream):
-        #     text_token = response.token.text
-
-        #     if response.details:
-        #         return
-
-        #     if text_token in STOP_SUSPECT_LIST:
-        #         acc_text += text_token
-        #         continue
-
-        #     if idx == 0 and text_token.startswith(" "):
-        #         text_token = text_token[1:]
-
-        #     acc_text += text_token
-        #     last_turn = list(chat_history.pop(-1))
-        #     last_turn[-1] += acc_text
-        #     chat_history = chat_history + [last_turn]
-        #     yield chat_history
-        #     acc_text = ""
 
     def delete_last_turn(chat_history):
         if chat_history: