Update app.py

app.py

@@ -253,22 +253,53 @@ with gr.Blocks(title=title) as demo:
 
     gr.Markdown("## Test Samples")
 
+    # with gr.Row():
+
+    #     # NORMAL
+    #     with gr.Column(scale=1):
+    #         gr.Markdown("### NORMAL")
+    #         gr.Image("images/NORMAL.jpeg", show_label=False, height=220)
+    
+    #     # VIRAL
+    #     with gr.Column(scale=1):
+    #         gr.Markdown("### VIRAL")
+    #         gr.Image("images/VIRAL.jpeg", show_label=False, height=220)
+    
+    #     # BACTERIAL
+    #     with gr.Column(scale=1):
+    #         gr.Markdown("### BACTERIAL")
+    #         gr.Image("images/BACT.jpeg", show_label=False, height=220)
     with gr.Row():
 
-        # NORMAL
         with gr.Column(scale=1):
             gr.Markdown("### NORMAL")
-            gr.Image("images/NORMAL.jpeg", show_label=False, height=220)
+            normal_sample = gr.Image("images/NORMAL.jpeg", show_label=False, height=220, interactive=True)
     
-        # VIRAL
         with gr.Column(scale=1):
             gr.Markdown("### VIRAL")
-            gr.Image("images/VIRAL.jpeg", show_label=False, height=220)
+            viral_sample = gr.Image("images/VIRAL.jpeg", show_label=False, height=220, interactive=True)
     
-        # BACTERIAL
         with gr.Column(scale=1):
             gr.Markdown("### BACTERIAL")
-            gr.Image("images/BACT.jpeg", show_label=False, height=220)
+            bact_sample = gr.Image("images/BACT.jpeg", show_label=False, height=220, interactive=True)
+
+    normal_sample.select(
+        inference_pipeline,
+        inputs=[normal_sample, thresh_b, thresh_v, seg_thresh],
+        outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
+    )
+    
+    viral_sample.select(
+        inference_pipeline,
+        inputs=[viral_sample, thresh_b, thresh_v, seg_thresh],
+        outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
+    )
+    
+    bact_sample.select(
+        inference_pipeline,
+        inputs=[bact_sample, thresh_b, thresh_v, seg_thresh],
+        outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
+    )
 
 if __name__ == "__main__":
     demo.launch()