Update app.py

app.py

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