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
|