Update app.py

app.py

@@ -133,9 +133,9 @@ def cellpose_segment(img_input):
     #masks = Image.fromarray(255. * crand[masks])
 
     pil_masks = Image.fromarray((masks/masks.max()).astype('float32'))
-    pil_masks.save("test.png")
+    pil_masks.save("test.tiff")
     
-    return outpix, overlay, flows, gr.DownloadButton(visible=True, value = "test.png")
+    return outpix, overlay, flows, gr.DownloadButton(visible=True, value = "test.tiff")
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -162,7 +162,7 @@ with gr.Blocks(title = "Hello",
             img_outlines = gr.Image(label = "Output image", type = "pil")            
             img_overlay = gr.Image(label = "Output image", type = "numpy")            
             flows = gr.Image(label = "Output image", type = "numpy")
-            masks = gr.Image(label = "Output image", type = "numpy")
+            #masks = gr.Image(label = "Output image", type = "numpy")
     
     
     send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[img_outlines, img_overlay, flows, down_btn])