Update app.py

app.py

@@ -154,12 +154,12 @@ def run_model_gpu(img, max_iter):
     return masks, flows
 
 @spaces.GPU(duration=60)
-def run_model_gpu60(img):
+def run_model_gpu60(img, max_iter):
     masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
     return masks, flows
 
 @spaces.GPU(duration=240)
-def run_model_gpu240(img):
+def run_model_gpu240(img, max_iter):
     masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
     return masks, flows
 
@@ -178,7 +178,7 @@ def cellpose_segment(filepath, resize = 1000,max_iter = 250):
             img = image_resize(img_input, resize = resize)
             
             maxsize = np.max(img.shape)
-            if maxsize<1000:
+            if maxsize<=1000:
                 masks, flows = run_model_gpu(img, max_iter=max_iter)
             elif maxsize < 5000:
                 masks, flows = run_model_gpu60(img, max_iter=max_iter)