Update app.py

app.py

@@ -227,6 +227,50 @@ title = "Chest X-ray: UNet segmentation + 2 binary classifiers"
 desc = "Pipeline: UNet -> mask lungs -> two binary classifiers (Normal vs Bacterial, Normal vs Viral). " \
        "If both classifiers fire, the stronger probability is chosen (fallback). Thresholds adjustable."
 
+with gr.Blocks() as demo:
+
+    gr.Markdown("### Example Table (Image + True Label)")
+
+    example_samples = [
+        ["images/NORMAL.jpg", "NORMAL"],
+        ["images/VIRAL.jpg", "VIRAL"],
+        ["images/BACT.jpg", "BACTERIAL"],
+    ]
+
+    gr.Dataset(
+        samples=example_samples,
+        components=[
+            gr.Image(type="filepath", label="Image"),
+            gr.Textbox(label="True Label")
+        ],
+        headers=["Image", "Label"],
+        label="Examples",
+        layout="table"
+    )
+
+    iface = gr.Interface(
+        fn=inference_pipeline,
+        inputs=[
+            gr.Image(type="numpy", label="Upload chest X-ray (RGB or grayscale)"),
+            gr.Slider(minimum=0.1, maximum=0.9, step=0.01, value=0.5, label="Bacterial threshold (thresh_b)"),
+            gr.Slider(minimum=0.1, maximum=0.9, step=0.01, value=0.5, label="Viral threshold (thresh_v)"),
+            gr.Slider(minimum=0.1, maximum=0.9, step=0.01, value=0.5, label="Segmentation mask threshold (seg_thresh)")
+        ],
+        outputs=[
+            gr.Label(num_top_classes=1, label="Prediction"),
+            gr.Number(label="Bacterial Probability"),
+            gr.Number(label="Viral Probability"),
+            gr.Image(type="pil", label="Masked Image (input × mask)"),
+            gr.Image(type="pil", label="Segmentation Overlay (red mask)")
+        ],
+        title=title,
+        description=desc,
+        allow_flagging="never"
+    )
+
+demo.launch(share=False)
+
+
 iface = gr.Interface(
     fn=inference_pipeline,
     inputs=[
@@ -242,11 +286,7 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Masked Image (input × mask)"),
         gr.Image(type="pil", label="Segmentation Overlay (red mask)")
     ],
-    gr.examples=[
-        ["images/NORMAL.jpg", 0.5, 0.5, 0.5,label="NORMAL"],
-        ["images/VIRAL.jpg", 0.5, 0.5, 0.5,label="VIRAL"],
-        ["images/BACT.jpg", 0.5, 0.5, 0.5,label="BACTERIAL"],
-    ],
+
     title=title,
     description=desc,
     allow_flagging="never"