Update app.py

app.py

@@ -147,7 +147,11 @@ def cellpose_segment(img_input):
 #    description="upload an image, then cellpose will segment it at a max size of 400x400 (for full functionality, 'pip install cellpose' locally)"
 #)
 
-   
+def download_function: 
+    b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
+    b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
+    return b1, b2
+
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
 
@@ -155,8 +159,10 @@ with gr.Blocks(title = "Hello",
         with gr.Column(scale=1):
             input_image = gr.Image(label = "Input image", type = "numpy")
             send_btn = gr.Button("Run Cellpose-SAM")
-            down_btn = gr.DownloadButton("Download masks as TIFF", visible=False)
-            down_btn2 = gr.DownloadButton("Download outline image as PNG", visible=False)
+            with gr.Column(scale=1):
+                down_btn = gr.DownloadButton("Download masks as TIFF", visible=False)
+            with gr.Column(scale=1):
+                down_btn2 = gr.DownloadButton("Download outline image as PNG", visible=False)
 
         with gr.Column(scale=1):            
             img_outlines = gr.Image(label = "Output image", type = "pil")            
@@ -167,7 +173,7 @@ with gr.Blocks(title = "Hello",
     
     send_btn.click(fn=cellpose_segment, inputs=[input_image], outputs=[img_outlines, img_overlay, flows, down_btn, down_btn2])
 
-    down_btn.click(None, None, None)
+    #down_btn.click(download_function, None, [down_btn, down_btn2])