mouseland commited on
Commit
8357e50
·
verified ·
1 Parent(s): f7e0b43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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