Fall back on avoiding '\nUser'

app.py

@@ -46,6 +46,11 @@ def chat_with_model(messages):
 
     for new_text in streamer:
         output_text += new_text
+        if "\nUser:" in output_text:
+            output_text = output_text.split("\nUser:")[0].rstrip()
+            messages[-1]["content"] = output_text
+            yield messages
+            break
         messages[-1]["content"] = output_text
         yield messages