Update app.py

app.py

@@ -114,9 +114,8 @@ def cellpose_segment(img_input):
     
     outpix = plot_outlines(img_input, masks)
     overlay = plot_overlay(img_input, masks)
-    masks =  Image.fromarray(masks)
     
-    return masks # outpix, overlay, flows, masks
+    return outpix, overlay, flows, masks
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -132,14 +131,15 @@ with gr.Blocks(title = "Hello",
 
     with gr.Row():
         input_image = gr.Image(label = "Input image", type = "numpy")
-        output_image3 = gr.Image(label = "Output image", type = "pil")
+        output_image1 = gr.Image(label = "Output image", type = "numpy")
 
-    #with gr.Row():
-    #    output_image2 = gr.Image(label = "Output image", type = "pil")
-    #    output_image3 = gr.Image(label = "Output image", type = "pil")
+    send_btn = gr.Button("Run Cellpose-SAM")
+    send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[output_image1, output_image2, output_image3])
+    
+    with gr.Row():
+        output_image2 = gr.Image(label = "Output image", type = "pil")
+        output_image3 = gr.Image(label = "Output image", type = "pil")
         
-    send_btn = gr.Button("Infer")
-    send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[output_image3])