Update app.py
Browse files
app.py
CHANGED
|
@@ -151,19 +151,16 @@ def image_resize(img, resize=400):
|
|
| 151 |
|
| 152 |
@spaces.GPU(duration=10)
|
| 153 |
def run_model_gpu(img, max_iter):
|
| 154 |
-
print(max_iter)
|
| 155 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 156 |
return masks, flows
|
| 157 |
|
| 158 |
@spaces.GPU(duration=60)
|
| 159 |
def run_model_gpu60(img, max_iter):
|
| 160 |
-
print(max_iter)
|
| 161 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 162 |
return masks, flows
|
| 163 |
|
| 164 |
@spaces.GPU(duration=240)
|
| 165 |
def run_model_gpu240(img, max_iter):
|
| 166 |
-
print(max_iter)
|
| 167 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 168 |
return masks, flows
|
| 169 |
|
|
|
|
| 151 |
|
| 152 |
@spaces.GPU(duration=10)
|
| 153 |
def run_model_gpu(img, max_iter):
|
|
|
|
| 154 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 155 |
return masks, flows
|
| 156 |
|
| 157 |
@spaces.GPU(duration=60)
|
| 158 |
def run_model_gpu60(img, max_iter):
|
|
|
|
| 159 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 160 |
return masks, flows
|
| 161 |
|
| 162 |
@spaces.GPU(duration=240)
|
| 163 |
def run_model_gpu240(img, max_iter):
|
|
|
|
| 164 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 165 |
return masks, flows
|
| 166 |
|