Update app.py

app.py

@@ -8,15 +8,25 @@ from skimage.measure import regionprops
 from skimage.segmentation import clear_border
 from cellpose import models
 
-# Load Cellpose model
-model = models.Cellpose(gpu=False, model_type="cyto")
+# Load Cellpose model - updated for v4.0.6
+try:
+    # Try the new API first
+    model = models.CellposeModel(gpu=False, model_type="cyto")
+except AttributeError:
+    # Fall back to older API if needed
+    model = models.Cellpose(gpu=False, model_type="cyto")
 
 def process_image(image):
     # Convert to numpy array and make grayscale
     image_np = np.array(image.convert("L"))
     
-    # Run Cellpose prediction
-    masks, flows, styles = model.eval(image_np, diameter=None, channels=[0,0])
+    try:
+        # Try new API call format
+        masks, flows, styles = model.eval(image_np, diameter=None, channels=[0,0])
+    except TypeError:
+        # Fall back to older API format
+        masks, flows, styles, diams = model.eval([image_np], diameter=None, channels=[0,0])
+        masks = masks[0]
     
     # Clean up borders and small artifacts
     masks_cleaned = clear_border(masks)