Update app.py

app.py

@@ -252,35 +252,24 @@ with gr.Blocks(title=title) as demo:
     )
 
 
-    test_samples = [
-        ("images/NORMAL.jpeg", "NORMAL"),
-        ("images/VIRAL.jpeg", "VIRAL"),
-        ("images/BACT.jpeg", "BACTERIAL"),
+    gr.Markdown("### Example Images with Ground Truth Labels")
+
+    example_samples = [
+        ["images/NORMAL.jpg", "NORMAL"],
+        ["images/VIRAL.jpg", "VIRAL"],
+        ["images/BACT.jpg", "BACTERIAL"],
     ]
-    
-    gr.Markdown("## Test Samples")
-    
-    sample_buttons = []
-    
-    with gr.Row():
-        for img_path, label in test_samples:
-            with gr.Column(scale=1):
-                gr.Markdown(f"### {label}")
-                btn = gr.Image(
-                    img_path,
-                    show_label=False,
-                    height=220,
-                    interactive=True
-                )
-                sample_buttons.append(btn)
-    
-    # Auto-run inference when any sample is clicked
-    for btn in sample_buttons:
-        btn.change(
-            inference_pipeline,
-            inputs=[btn, thresh_b, thresh_v, seg_thresh],
-            outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
-        )
+
+    gr.Dataset(
+        samples=example_samples,
+        components=[
+            gr.Image(type="filepath", label="Image", interactive=False),
+            gr.Markdown(label="True Label")
+        ],
+        headers=["Image", "True Label"],
+        label="Examples Table",
+        layout="table"
+    )
 
 
 if __name__ == "__main__":