semmyk commited on
Commit
dd19d58
·
1 Parent(s): bc8a2ce

baseline08_beta0.3.6_03Oct25: fixing slow Marker: @spaces.GPU duration int err

Browse files
Files changed (1) hide show
  1. ui/gradio_ui.py +2 -1
ui/gradio_ui.py CHANGED
@@ -261,7 +261,8 @@ def convert_batch(
261
  #progress((10,16), desc=f"ProcessPoolExecutor: Pooling file conversion result: [{str(result_interim)}[:20]]")
262
  #progress2((10,16), desc=f"ProcessPoolExecutor: Pooling file conversion result: [{str(result_interim)}[:20]]")
263
  #time.sleep(0.25)'''
264
- @spaces.GPU(duration=3.75*len(pdf_files_count)) ## HF Spaces GPU support
 
265
  def get_results_pool_map(pdf_files, pdf_files_count, progress2=gr.Progress()):
266
  #Use progress.tqdm to integrate with the executor map
267
  #results = pool.map(pdf2md_converter.convert_files, pdf_files) ##SMY iterables #max_retries #output_dir_string)
 
261
  #progress((10,16), desc=f"ProcessPoolExecutor: Pooling file conversion result: [{str(result_interim)}[:20]]")
262
  #progress2((10,16), desc=f"ProcessPoolExecutor: Pooling file conversion result: [{str(result_interim)}[:20]]")
263
  #time.sleep(0.25)'''
264
+ duration = 5.75 * len(pdf_files_count) if len(pdf_files_count)>=2 else 7
265
+ @spaces.GPU(duration=duration) ## HF Spaces GPU support
266
  def get_results_pool_map(pdf_files, pdf_files_count, progress2=gr.Progress()):
267
  #Use progress.tqdm to integrate with the executor map
268
  #results = pool.map(pdf2md_converter.convert_files, pdf_files) ##SMY iterables #max_retries #output_dir_string)