fix gradio demo stripping characters

10067 · Mar 4, 2024 · 1f86276 · 1f86276
1 parent a8f5d63
commit 1f86276
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 .venv
-__pycache__
+__pycache__
+checkpoints
diff --git a/gradio_demo.py b/gradio_demo.py
@@ -42,9 +42,9 @@ def answer_question(img, prompt):
 
     buffer = ""
     for new_text in streamer:
-        clean_text = re.sub("<$|END$", "", new_text)
+        clean_text = re.sub("<$|<END$", "", new_text)
         buffer += clean_text
-        yield buffer.strip("<END")
+        yield buffer
 
 
 with gr.Blocks() as demo: