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":
|