Update app.py

GPU should only be active for model eval

app.py

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