Update app.py

app.py

@@ -250,24 +250,25 @@ with gr.Blocks(title=title) as demo:
         inputs=[img_in, thresh_b, thresh_v, seg_thresh],
         outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
     )
-    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", interactive=False),
-        #     gr.Markdown(label="True Label")
-        # ],
-        headers=["Image", "True Label"],
-        # label="Examples Table",
-        layout="table"
-    )
 
+    gr.Markdown("## Test Samples")
+
+    with gr.Row():
 
+        # NORMAL
+        with gr.Column(scale=1):
+            gr.Markdown("### NORMAL")
+            gr.Image("images/NORMAL.jpeg", show_label=False, height=220)
+    
+        # VIRAL
+        with gr.Column(scale=1):
+            gr.Markdown("### VIRAL")
+            gr.Image("images/VIRAL.jpeg", show_label=False, height=220)
+    
+        # BACTERIAL
+        with gr.Column(scale=1):
+            gr.Markdown("### BACTERIAL")
+            gr.Image("images/BACT.jpeg", show_label=False, height=220)
 
 if __name__ == "__main__":
     demo.launch()