Update app.py

app.py

@@ -250,28 +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]
     )
-    img_normal_path = "images/NORMAL.jpeg"
-    img_viral_path = "images/VIRAL.jpeg"
-    img_bacterial_path = "images/BACT.jpeg"
-    table_data = [
-        [img_normal_path, "NORMAL"],
-        [img_viral_path, "VIRAL"],
-        [img_bacterial_path, "BACTERIAL"]
+    example_samples = [
+        ["images/NORMAL.jpg", "NORMAL"],
+        ["images/VIRAL.jpg", "VIRAL"],
+        ["images/BACT.jpg", "BACTERIAL"],
     ]
 
-    gr.Markdown("### ☰ Test Samples")
-    
-    gr.Dataframe(
-        value=table_data,
-        headers=["Image", "Label1"],
-        datatype=["image", "str"],
-        show_label=False,
-        interactive=False,
-        min_height=350,  
-        wrap=True        
+    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"
     )
 
 
+
 if __name__ == "__main__":
     demo.launch()