a

app.py

@@ -569,13 +569,23 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                 with gr.Column():
                     image_output = gr.Image(label="Segmented Image", type="pil", height=400)
             
+                # Button to trigger the prediction
+                predict_button = gr.Button("Predict Mask")
+                
+                # Define the action triggered by the predict button
+                predict_button.click(
+                    fn=predict_masks,
+                    inputs=[upload_image_input],
+                    outputs=[image_output],
+                    show_progress=True,
+                )
             # Define the action triggered by the upload_image_input change
-            upload_image_input.change(
-                fn=update_mask,
-                inputs=[upload_image_input],
-                outputs=[image_output],
-                show_progress=True,
-            )
+            # upload_image_input.change(
+            #     fn=update_mask,
+            #     inputs=[upload_image_input],
+            #     outputs=[image_output],
+            #     show_progress=True,
+            # )
         with gr.TabItem("Misc"):
             with gr.Column():
                 clear_cache_button = gr.Button("Clear CUDA Cache")