Update app.py

app.py

@@ -114,8 +114,9 @@ def cellpose_segment(img_input):
     
     outpix = plot_outlines(img_input, masks)
     overlay = plot_overlay(img_input, masks)
-    
-    return outpix, overlay, flows, masks
+
+    iperm = np.random.permutation(np.max(masks.flatten()).astype('int')+1)
+    return outpix, overlay, flows, iperm[masks]
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -135,12 +136,13 @@ with gr.Blocks(title = "Hello",
             send_btn = gr.Button("Run Cellpose-SAM")
 
         with gr.Column(scale=1):
-            output_image1 = gr.Image(label = "Output image", type = "numpy")
-            output_image2 = gr.Image(label = "Output image", type = "numpy")
-            output_image3 = gr.Image(label = "Output image", type = "numpy")
-            output_image4 = gr.Image(label = "Output image", type = "numpy")
+            img_outlines = gr.Image(label = "Output image", type = "numpy")
+            masks = gr.Image(label = "Output image", type = "numpy")
+            img_overlay = gr.Image(label = "Output image", type = "numpy")
+            flows = gr.Image(label = "Output image", type = "numpy")
+            
 
-    send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[output_image1, output_image4, output_image2, output_image3])
+    send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[img_outlines, img_overlay, flows, masks])