Update app.py

app.py

@@ -241,6 +241,15 @@ with gr.Blocks(title=title) as demo:
             submit_btn = gr.Button("Submit", variant="primary")   
             clear_btn = gr.Button("Clear", variant="secondary") 
 
+            
+
+        with gr.Column():
+            pred_out = gr.Label(num_top_classes=1, label="Prediction")
+            pb_out = gr.Number(label="Bacterial Probability")
+            pv_out = gr.Number(label="Viral Probability")
+            masked_img_out = gr.Image(type="pil", label="Masked Image")
+            overlay_out = gr.Image(type="pil", label="Segmentation Overlay")
+
             submit_btn.click(
                 inference_pipeline,
                 inputs=[img_in, thresh_b, thresh_v, seg_thresh],
@@ -252,13 +261,6 @@ with gr.Blocks(title=title) as demo:
                 outputs=[img_in, pred_out, pb_out, pv_out, masked_img_out, overlay_out]
             )
 
-        with gr.Column():
-            pred_out = gr.Label(num_top_classes=1, label="Prediction")
-            pb_out = gr.Number(label="Bacterial Probability")
-            pv_out = gr.Number(label="Viral Probability")
-            masked_img_out = gr.Image(type="pil", label="Masked Image")
-            overlay_out = gr.Image(type="pil", label="Segmentation Overlay")
-
     img_in.change(
         inference_pipeline,
         inputs=[img_in, thresh_b, thresh_v, seg_thresh],