Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -161,7 +161,8 @@ def process_document_stream(model_name: str, task_choice: str, image: Image.Imag
|
|
| 161 |
|
| 162 |
thread = Thread(target=model.generate, kwargs=generation_kwargs)
|
| 163 |
thread.start()
|
| 164 |
-
|
|
|
|
| 165 |
# 4. Stream raw output to the UI in real-time
|
| 166 |
buffer = ""
|
| 167 |
for new_text in streamer:
|
|
@@ -169,7 +170,6 @@ def process_document_stream(model_name: str, task_choice: str, image: Image.Imag
|
|
| 169 |
buffer = buffer.replace("<|im_end|>", "")
|
| 170 |
time.sleep(0.01)
|
| 171 |
yield buffer
|
| 172 |
-
yield progress_bar_html("streaming markdown") # Show progress bar first
|
| 173 |
#yield buffer , "⏳ Processing...", {"status": "streaming"}
|
| 174 |
|
| 175 |
# 5. Post-process the final buffer based on the selected task
|
|
|
|
| 161 |
|
| 162 |
thread = Thread(target=model.generate, kwargs=generation_kwargs)
|
| 163 |
thread.start()
|
| 164 |
+
|
| 165 |
+
yield progress_bar_html("streaming markdown") # Show progress bar first
|
| 166 |
# 4. Stream raw output to the UI in real-time
|
| 167 |
buffer = ""
|
| 168 |
for new_text in streamer:
|
|
|
|
| 170 |
buffer = buffer.replace("<|im_end|>", "")
|
| 171 |
time.sleep(0.01)
|
| 172 |
yield buffer
|
|
|
|
| 173 |
#yield buffer , "⏳ Processing...", {"status": "streaming"}
|
| 174 |
|
| 175 |
# 5. Post-process the final buffer based on the selected task
|