Spaces:
Running on Zero
Running on Zero
update app
Browse files
app.py
CHANGED
|
@@ -580,7 +580,7 @@ def predictions_to_rrd(predictions, glbfile, target_dir, frame_filter="All", sho
|
|
| 580 |
return rrd_path
|
| 581 |
|
| 582 |
|
| 583 |
-
@spaces.GPU(duration=120)
|
| 584 |
def run_model(target_dir, apply_mask=True, mask_edges=True, filter_black_bg=False, filter_white_bg=False):
|
| 585 |
global model
|
| 586 |
import torch
|
|
|
|
| 580 |
return rrd_path
|
| 581 |
|
| 582 |
|
| 583 |
+
@spaces.GPU(duration=120, size="xlarge")
|
| 584 |
def run_model(target_dir, apply_mask=True, mask_edges=True, filter_black_bg=False, filter_white_bg=False):
|
| 585 |
global model
|
| 586 |
import torch
|