Fix end on "User:"

app.py

@@ -92,8 +92,8 @@ def chat_with_model(messages):
             streamer.end_of_generation.set()  # signal to stop generation thread
             break
 
-        if "\nUser:" in token_str:
-            break
+        if is_special:
+            continue
 
         if "<think>" in token_str:
             in_think = True
@@ -107,6 +107,9 @@ def chat_with_model(messages):
         else:
             output_text += token_str
 
+        if "\nUser:" in output_text:
+            output_text = output_text.split("\nUser:")[0].rstrip()
+
         messages[-1]["content"] = output_text
         yield messages