Update app.py

app.py

@@ -251,21 +251,21 @@ with gr.Blocks(title=title) as demo:
         outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
     )
 
-    example_samples = [
+    table_data = [
         ["images/NORMAL.jpg", "NORMAL"],
         ["images/VIRAL.jpg", "VIRAL"],
         ["images/BACT.jpg", "BACTERIAL"],
     ]
 
-    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"
+    gr.Markdown("### ☰ Test Samples")
+    
+    gr.Dataframe(
+        value=table_data,
+        headers=["Image", "Label"],
+        datatype=["image", "str"],
+        show_label=False,   
+        interactive=False,    
+        height=350            
     )