Update main.py

main.py

@@ -115,7 +115,7 @@ def predict(message: str, history: list[str]) -> str:
             partial_message += new_token
             yield partial_message
 
-    yield bypass
+    yield partial_message + bypass
 
 
 # Create and run the gradio interface