fffiloni commited on
Commit
a861018
·
verified ·
1 Parent(s): 26ca839

enable tqdm tracking

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ def run_inference(temp_dir, removed_bg_path):
132
  except subprocess.CalledProcessError as e:
133
  return f"Error during inference: {str(e)}"
134
 
135
- def process_image(input_pil, remove_bg):
136
 
137
  torch.cuda.empty_cache()
138
 
 
132
  except subprocess.CalledProcessError as e:
133
  return f"Error during inference: {str(e)}"
134
 
135
+ def process_image(input_pil, remove_bg, progress=gr.Progress(track_tqdm=True)):
136
 
137
  torch.cuda.empty_cache()
138