update

app.py

@@ -104,14 +104,13 @@ def cellpose_segment(img):
 
     outpix = plot_outlines(img, masks)
     overlay = plot_overlay(img, masks)
-    print(outpix.shape)
-    return overlay, flows, masks 
+    return outpix, overlay, flows, masks 
 
 # Gradio Interface
 iface = gr.Interface(
     fn=cellpose_segment, 
     inputs="image", 
-    outputs=["image", "image", "image"],
+    outputs=["image", "image", "image", "image"],
     title="cellpose segmentation",
     description="upload an image, then cellpose will segment it at a max size of 224x224"
 )