Update app.py

app.py

@@ -227,49 +227,6 @@ 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,
@@ -291,6 +248,24 @@ iface = gr.Interface(
     description=desc,
     allow_flagging="never"
 )
+gr.Markdown("### Example Images with True Labels")
+
+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="Test Samples",
+    layout="table"
+)
 
 if __name__ == "__main__":
     iface.launch(share=False)