Spaces:
Running
on
L40S
Running
on
L40S
enable tqdm tracking
Browse files
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 |
|