Update app.py

app.py

@@ -158,7 +158,7 @@ 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, max_iter)
+@spaces.GPU(duration=240)
 def run_model_gpu240(img, max_iter):
     masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
     return masks, flows
@@ -179,11 +179,11 @@ def cellpose_segment(filepath, resize = 1000,max_iter = 250):
             
             maxsize = np.max(img.shape)
             if maxsize<=1000:
-                masks, flows = run_model_gpu(img, max_iter=max_iter)
+                masks, flows = run_model_gpu(img, max_iter)
             elif maxsize < 5000:
-                masks, flows = run_model_gpu60(img, max_iter=max_iter)
+                masks, flows = run_model_gpu60(img, max_iter)
             elif maxsize < 20000:
-                masks, flows = run_model_gpu240(img, max_iter=max_iter)
+                masks, flows = run_model_gpu240(img, max_iter)
             else:
                 raise ValueError("Image size must be less than 20,000")