shape

app.py

@@ -102,8 +102,8 @@ def cellpose_segment(img_input):
         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)
+    outpix = plot_outlines(img_input, masks)
+    overlay = plot_overlay(img_input, masks)
     return outpix, overlay, flows, masks 
 
 # Gradio Interface