prithivMLmods commited on
Commit
7f05ca0
·
verified ·
1 Parent(s): 171dbfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- 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:
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":