Update app.py

app.py

@@ -228,14 +228,14 @@ desc = "Pipeline: UNet -> mask lungs -> two binary classifiers (Normal vs Bacter
        "If both classifiers fire, the stronger probability is chosen (fallback). Thresholds adjustable."
 
 
+# Table data (Images embedded as HTML)
 example_samples = [
-    ("images/NORMAL.jpeg", "NORMAL"),
-    ("images/VIRAL.jpeg", "VIRAL"),
-    ("images/BACT.jpeg", "BACTERIAL"),
+    ['images/NORMAL.jpeg',"NORMAL"],
+    ['images/VIRAL.jpeg',"VIRAL"],
+    ['images/BACT.jpeg',"BACTERIAL"],
 ]
 
 with gr.Blocks(title=title) as demo:
-
     gr.Markdown(f"### {title}")
     gr.Markdown(desc)
 
@@ -259,16 +259,14 @@ with gr.Blocks(title=title) as demo:
 
     gr.Markdown("### Test Samples")
 
-    # Table Header
-    with gr.Row():
-        gr.Markdown("**Image**")
-        gr.Markdown("**Label**")
-
-    # Table Rows (Images + Labels)
-    for img_path, label in example_samples:
-        with gr.Row():
-            gr.Image(img_path, width=150, show_label=False)
-            gr.Markdown(f"<center>{label}</center>")
+    gr.Dataframe(
+        headers=["Image", "Label"],
+        value=example_samples,
+        interactive=False,
+        wrap=True,
+        row_count=(len(example_samples), "fixed"),
+        col_count=(2, "fixed")
+    )
 
 if __name__ == "__main__":
     demo.launch(share=False)