Update app.py

app.py

@@ -99,16 +99,16 @@ def image_resize(img, resize=400):
     return img
 
     
-#@spaces.GPU(duration=10)
-#def run_model_gpu(model, img):
-#    masks, flows, _ = model.eval(img, channels=[0,0])
- #   return masks, flows
-
 @spaces.GPU(duration=10)
+def run_model_gpu(img):
+    masks, flows, _ = model.eval(img, channels=[0,0])
+    return masks, flows
+
+#@spaces.GPU(duration=10)
 def cellpose_segment(img_input):
     img = image_resize(img_input)
-    #masks, flows = run_model_gpu(model, img)
-    masks, flows, _ = model.eval(img, channels=[0,0])
+    masks, flows = run_model_gpu(img)
+    #masks, flows, _ = model.eval(img, channels=[0,0])
     flows = flows[0]
     # masks = np.zeros(img.shape[:2])
     # flows = np.zeros_like(img)