diff --git a/app.py b/app.py index f467107..b83d6a0 100644 --- a/app.py +++ b/app.py @@ -114,9 +114,10 @@ def model_inference( doctag_output = "" for new_text in streamer: - if new_text != "": - buffer += html.escape(new_text) - doctag_output += new_text + if "" in new_text: + continue # Skip it entirely + buffer += html.escape(new_text) + doctag_output += new_text yield buffer if any(tag in doctag_output for tag in ["", "", "", "", ""]):