updating with resize

app.py

@@ -90,18 +90,18 @@ def image_resize(img, resize=224):
     img = img.astype(np.uint8)
     return img
 
-def cellpose_segment(img):
-    img_input = image_resize(img.copy())
-    masks, flows, _ = model.eval(img_input, channels=[0,0])
+def cellpose_segment(img_input):
+    img = image_resize(img_input)
+    masks, flows, _ = model.eval(img, channels=[0,0])
     flows = flows[0]
     # masks = np.zeros(img.shape[:2])
     # flows = np.zeros_like(img)
     target_size = (img_input.shape[1], img_input.shape[0])
-    if (target_size[0]!=img.shape[1] and target_size[1]!=img.shape[0]):
+    if (target_size[0]!=img.shape[1] or target_size[1]!=img.shape[0]):
         # scale it back to keep the orignal size
         masks = cv2.resize(masks.astype('uint16'), target_size, interpolation=cv2.INTER_NEAREST).astype('uint16')
         flows = cv2.resize(flows.astype('float32'), target_size).astype('uint8')
-
+    
     outpix = plot_outlines(img, masks)
     overlay = plot_overlay(img, masks)
     return outpix, overlay, flows, masks