Update app.py

app.py

@@ -232,8 +232,13 @@ def upload_file(filepath):
     img = imread(filepath)    
     img = normalize99(img)
     img = np.clip(img, 0, 1)
+
+    b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
+    b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
+    return img, filepath, b1, b2
+
+def upload_example():
     
-    return img, filepath
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -269,9 +274,9 @@ with gr.Blocks(title = "Hello",
     sample_list = []
     for j in range(23):
         sample_list.append("samples/img%0.2d.png"%j)
-    gr.Examples(sample_list, inputs=input_image, examples_per_page=25)
+    gr.Examples(sample_list, inputs=up_btn, examples_per_page=25)
     
-    up_btn.upload(upload_file, up_btn, [input_image, up_btn])
+    up_btn.upload(upload_file, up_btn, [input_image, up_btn, down_btn, down_btn2])
     send_btn.click(cellpose_segment, [up_btn, resize], [img_outlines, img_overlay, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])