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 |
|