Spaces:
Running
on
Zero
Running
on
Zero
increase ZeroGPU needed time for inference
Browse files
app.py
CHANGED
|
@@ -133,7 +133,7 @@ def run_inference(temp_dir, removed_bg_path):
|
|
| 133 |
except subprocess.CalledProcessError as e:
|
| 134 |
return f"Error during inference: {str(e)}"
|
| 135 |
|
| 136 |
-
@spaces.GPU(duration=
|
| 137 |
def process_image(input_pil, remove_bg, progress=gr.Progress(track_tqdm=True)):
|
| 138 |
|
| 139 |
torch.cuda.empty_cache()
|
|
|
|
| 133 |
except subprocess.CalledProcessError as e:
|
| 134 |
return f"Error during inference: {str(e)}"
|
| 135 |
|
| 136 |
+
@spaces.GPU(duration=140)
|
| 137 |
def process_image(input_pil, remove_bg, progress=gr.Progress(track_tqdm=True)):
|
| 138 |
|
| 139 |
torch.cuda.empty_cache()
|