Update app.py

app.py

@@ -231,7 +231,11 @@ with gr.Blocks(title = "Hello",
             gr.HTML("""<h4 style="color:white;">"pip install cellpose" for full functionality. </h4>""")
 
             input_image = gr.Image(label = "Input image", type = "filepath")
-            send_btn = gr.Button("Run Cellpose-SAM")
+
+            with gr.Row():
+                resize = gr.Number(label = 'max size', value = 400)
+                send_btn = gr.Button("Run Cellpose-SAM")
+                
             with gr.Row():
                 down_btn = gr.DownloadButton("Download masks (TIF)", visible=False)            
                 down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)
@@ -247,7 +251,7 @@ with gr.Blocks(title = "Hello",
             #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, down_btn2])
+    send_btn.click(fn=cellpose_segment, inputs=[input_image, resize], outputs=[img_outlines, img_overlay, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])