bug fix

app.py

@@ -2,6 +2,7 @@ import math
 import os
 
 import gradio as gr
+from datasets import Image as HFImage
 from datasets import load_dataset
 
 REPO_ID = "teddyk251/self-imagine-sample"  # change me
@@ -9,6 +10,15 @@ TOKEN = os.environ.get("HF_TOKEN")
 
 ds = load_dataset(REPO_ID, split="train", token=TOKEN)
 
+ds = ds.cast_column("image", HFImage(decode=False))
+
+def img_src(img):
+    # HF Datasets gives either a dict with 'path' (local cache) or 'url'
+    if isinstance(img, dict):
+        return img.get("path") or img.get("url")
+    return img  # already a string
+
+
 DATASETS = ["All"] + sorted(set(ds["dataset"]))
 MODELS = ["All"] + sorted(set(ds["model"]))
 
@@ -30,8 +40,15 @@ def paginate(rows, page, page_size):
     start = (page - 1) * page_size
     end = min(start + page_size, total)
     sub = rows.select(range(start, end))
-    gallery_items = [(sub[i]["image"], f'{sub[i]["dataset"]} / {sub[i]["model"]} — {sub[i]["filename"]}')
-                     for i in range(len(sub))]
+    # gallery_items = [(sub[i]["image"], f'{sub[i]["dataset"]} / {sub[i]["model"]} — {sub[i]["filename"]}')
+    #                  for i in range(len(sub))]\
+    gallery_items = []
+    for i in range(len(sub)):
+       src = img_src(sub[i]["image"])
+       if not src:
+           continue  # skip records without a resolvable path/url
+       label = f'{sub[i]["dataset"]} / {sub[i]["model"]} — {sub[i].get("filename","")}'
+       gallery_items.append((src, label))
     return gallery_items, total, total_pages, page
 
 def refresh(dataset_sel, model_sel, query, page, page_size):
@@ -82,4 +99,4 @@ with gr.Blocks(title="Self‑Imagine — Sample Gallery") as demo:
     next_btn.click(on_next, [dataset_dd, model_dd, query_tb, page_num, page_size], [gallery, page_info, page_num])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(ssr_mode=False)

requirements.txt

@@ -1,2 +1,4 @@
 gradio>=4.30
 datasets>=2.19
+pillow
+pandas