Update app.py

app.py

@@ -22,6 +22,24 @@ def health_check():
 
 
 
+REPO_ID = "rahul7star/ohamlab"
+BASE_URL = f"https://huggingface.co/{REPO_ID}/resolve/main/demo/"
+
+@app.get("/images")
+def list_images():
+    try:
+        all_files = list_repo_files(REPO_ID)
+        demo_files = [f for f in all_files if f.startswith("demo/")]
+
+        image_files = [
+            BASE_URL + f[len("demo/"):]
+            for f in demo_files
+            if f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
+        ]
+
+        return {"images": image_files}
+    except Exception as e:
+        return {"error": str(e)}
 
 
 from datetime import datetime