Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image, ImageOps
 import numpy as np
 import gradio as gr
 import os
+import base64
 
 class DoubleConv(nn.Module):
     def __init__(self, in_channels, out_channels):
@@ -228,11 +229,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)
+def encode_image(img_path):
+    with open(img_path, "rb") as img:
+        return "data:image/png;base64," + base64.b64encode(img.read()).decode()
+
 example_samples = [
-    ['images/NORMAL.jpeg',"NORMAL"],
-    ['images/VIRAL.jpeg',"VIRAL"],
-    ['images/BACT.jpeg',"BACTERIAL"],
+    [f"<img src='{encode_image('images/NORMAL.jpeg')}' width='120'>", "NORMAL"],
+    [f"<img src='{encode_image('images/VIRAL.jpeg')}' width='120'>", "VIRAL"],
+    [f"<img src='{encode_image('images/BACT.jpeg')}' width='120'>", "BACTERIAL"],
 ]
 
 with gr.Blocks(title=title) as demo: