Update app.py

app.py

@@ -139,8 +139,8 @@ def cellpose_segment(img_input):
     # masks = np.zeros(img.shape[:2])
     # flows = np.zeros_like(img)
 
-    outpix = plot_outlines(img_input, masks)
-    overlay = plot_overlay(img_input, masks)
+    outpix = plot_outlines(img, masks)
+    overlay = plot_overlay(img, masks)
     
     target_size = (img_input.shape[1], img_input.shape[0])
     if (target_size[0]!=img.shape[1] or target_size[1]!=img.shape[0]):