Update app.py
This commit is contained in:
parent
0511dd38cc
commit
4ff2750a5c
7
app.py
7
app.py
|
|
@ -114,10 +114,9 @@ def model_inference(
|
||||||
doctag_output = ""
|
doctag_output = ""
|
||||||
|
|
||||||
for new_text in streamer:
|
for new_text in streamer:
|
||||||
if "<end_of_utterance>" in new_text:
|
if new_text != "<end_of_utterance>":
|
||||||
continue # Skip it entirely
|
buffer += html.escape(new_text)
|
||||||
buffer += html.escape(new_text)
|
doctag_output += new_text
|
||||||
doctag_output += new_text
|
|
||||||
yield buffer
|
yield buffer
|
||||||
|
|
||||||
if any(tag in doctag_output for tag in ["<doctag>", "<otsl>", "<code>", "<formula>", "<chart>"]):
|
if any(tag in doctag_output for tag in ["<doctag>", "<otsl>", "<code>", "<formula>", "<chart>"]):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue