Update app.py

app.py

@@ -237,6 +237,20 @@ with gr.Blocks(title=title) as demo:
             thresh_b = gr.Slider(0.1, 0.9, 0.5, step=0.01, label="Bacterial threshold")
             thresh_v = gr.Slider(0.1, 0.9, 0.5, step=0.01, label="Viral threshold")
             seg_thresh = gr.Slider(0.1, 0.9, 0.5, step=0.01, label="Segmentation mask threshold")
+        
+            submit_btn = gr.Button("Submit", variant="primary")   
+            clear_btn = gr.Button("Clear", variant="secondary") 
+
+            submit_btn.click(
+                inference_pipeline,
+                inputs=[img_in, thresh_b, thresh_v, seg_thresh],
+                outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
+            )
+            
+            clear_btn.click(
+                lambda: (None, None, None, None, None, None),
+                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")