Update app.py
Browse files
app.py
CHANGED
|
@@ -158,7 +158,7 @@ def run_model_gpu60(img, max_iter):
|
|
| 158 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 159 |
return masks, flows
|
| 160 |
|
| 161 |
-
@spaces.GPU(duration=240)
|
| 162 |
def run_model_gpu240(img, max_iter):
|
| 163 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 164 |
return masks, flows
|
|
|
|
| 158 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 159 |
return masks, flows
|
| 160 |
|
| 161 |
+
@spaces.GPU(duration=240, max_iter)
|
| 162 |
def run_model_gpu240(img, max_iter):
|
| 163 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 164 |
return masks, flows
|