chore: remove button and inference on image loading

app.py

@@ -480,11 +480,11 @@ def handle_upload_preview(
 
         return (
             display,
-            "Image uploaded. Click 'Run inference' to compute predictions.",
-            "",  # Reset prediction text
+            "Image uploaded. Computing predictions...",
+            "",
             pd.DataFrame(),
             gr.update(value=""),
-            {"image": np_image, "mask": None},  # Store raw image for inference
+            {"image": np_image, "mask": None},
         )
     except Exception as exc:  # pragma: no cover - surfaced in UI
         return None, f"Failed to load image: {exc}", "", pd.DataFrame(),  gr.update(value=""), None
@@ -547,7 +547,6 @@ def build_demo() -> gr.Blocks:
 
         gr.Markdown("---")
 
-        infer_btn = gr.Button("Run inference", variant="primary")
         status_text = gr.Markdown()
         with gr.Row():
             with gr.Column():
@@ -592,6 +591,10 @@ def build_demo() -> gr.Blocks:
                 ground_truth_display,
                 image_state,
             ],
+        ).then(
+            fn=run_inference,
+            inputs=[image_state, head_dropdown],
+            outputs=[main_prediction, prediction_probs],
         )
 
         upload_component.upload(
@@ -605,9 +608,7 @@ def build_demo() -> gr.Blocks:
                 ground_truth_display,
                 image_state,
             ],
-        )
-
-        infer_btn.click(
+        ).then(
             fn=run_inference,
             inputs=[image_state, head_dropdown],
             outputs=[main_prediction, prediction_probs],