Update app.py

app.py

@@ -150,22 +150,22 @@ def image_resize(img, resize=400):
     
 @spaces.GPU(duration=10)
 def run_model_gpu(img):
-    masks, flows, _ = model.eval(img)#, channels = [0,0])
+    masks, flows, _ = model.eval(img, niter = 250)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=60)
 def run_model_gpu60(img):
-    masks, flows, _ = model.eval(img)#, channels = [0,0])
+    masks, flows, _ = model.eval(img, niter = 500)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=240)
 def run_model_gpu240(img):
-    masks, flows, _ = model.eval(img)#, channels = [0,0])
+    masks, flows, _ = model.eval(img, niter = 500)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=1000)
 def run_model_gpu1000(img):
-    masks, flows, _ = model.eval(img)#, channels = [0,0])
+    masks, flows, _ = model.eval(img, niter = 500)#, channels = [0,0])
     return masks, flows