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