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

Update app.py

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