Update app.py

app.py

@@ -4,10 +4,28 @@ from src.utils import load_experiment
 from interpretability_demo import generate_rollout_for_demo
 from data.transforms import build_coco_transform
 
+from huggingface_hub import snapshot_download
+import os
+
+def load_model():
+    repo_id = "evanec/coco-demo"
+    local_dir = snapshot_download(repo_id)
+
+    base_path = os.path.join(
+        local_dir,
+        "experiment_54_clipencoder_last4_t5base_full",
+        "20251204_140137"
+    )
+
+
+    return base_path
+
+
 
 # Load Model + Preprocessing
 
-CHECKPOINT = "experiment_21_full_vlm_ceiling/20251127_231732"
+CHECKPOINT = load_model()
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model, tokenizer, meta, config = load_experiment(CHECKPOINT, device=device)