Create app.py

app.py

@@ -0,0 +1,25 @@
+import os
+import torch
+from huggingface_hub import hf_hub_download
+
+MODEL_REPO_ID = "farrell236/CephVIT"
+MODEL_FILENAME = "best.pt"
+
+hf_token = os.getenv("HF_TOKEN")
+assert hf_token, "HF_TOKEN missing"
+
+local_path = hf_hub_download(
+    repo_id=MODEL_REPO_ID,
+    filename=MODEL_FILENAME,
+    token=hf_token,
+)
+
+print("Downloaded to:", local_path)
+
+ckpt = torch.load(local_path, map_location="cpu")
+print("Type:", type(ckpt))
+
+if isinstance(ckpt, dict):
+    print("Top-level keys:", list(ckpt.keys())[:20])
+    if "state_dict" in ckpt and isinstance(ckpt["state_dict"], dict):
+        print("state_dict keys:", list(ckpt["state_dict"].keys())[:20])