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

baseline08_beta0.3.5_03Oct25: fixing slow Marker: @spaces.GPU duration = 3.75*pdf_files_count

Browse files
Files changed (1) hide show
  1. ui/gradio_ui.py +1 -1
ui/gradio_ui.py CHANGED
@@ -261,7 +261,7 @@ 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 ## 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
+ @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)