Update app.py

app.py

@@ -116,15 +116,31 @@ def cellpose_segment(img_input):
     overlay = plot_overlay(img_input, masks)
     masks =  Image.fromarray(masks)
     
-    return outpix, overlay, flows, masks
+    return masks # outpix, overlay, flows, masks
 
 # Gradio Interface
-iface = gr.Interface(
-    fn=cellpose_segment, 
-    inputs="image", 
-    outputs=["image", "image", "image", "image"],
-    title="cellpose segmentation",
-    description="upload an image, then cellpose will segment it at a max size of 400x400 (for full functionality, 'pip install cellpose' locally)"
-)
+#iface = gr.Interface(
+#    fn=cellpose_segment, 
+#    inputs="image", 
+#    outputs=["image", "image", "image", "image"],
+#    title="cellpose segmentation",
+#    description="upload an image, then cellpose will segment it at a max size of 400x400 (for full functionality, 'pip install cellpose' locally)"
+#)
 
-iface.launch()
+with gr.Blocks(title = "Hello", 
+               css=".gradio-container {background:purple;}") as demo:
+
+    with gr.Row():
+        input_image = gr.Image(label = "Input image", type = "pil")
+        output_image3 = gr.Image(label = "Output image", type = "pil")
+
+    #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=infer, inputs=[model, input_image], outputs=[output_image3])
+    
+    
+
+demo.launch()