Updated the checkpoint path

app.py

@@ -1,6 +1,17 @@
-from huggingface_hub import hf_hub_download
-ckpt_path = hf_hub_download(repo_id="c1tr0n75/VoxelPathFinder",
-                            filename="training_outputs/final_model.pth")
+from huggingface_hub import snapshot_download
+import os, glob, torch
+
+local_dir = snapshot_download(repo_id="c1tr0n75/VoxelPathFinder")
+
+# preferred path in subfolder
+ckpt_path = os.path.join(local_dir, "training_outputs", "final_model.pth")
+if not os.path.exists(ckpt_path):
+  # fallback: find any .pth
+  pths = [p for p in glob.glob(os.path.join(local_dir, "**", "*.pth"), recursive=True)]
+  if not pths:
+    raise FileNotFoundError("No .pth found in repo snapshot.")
+  ckpt_path = pths[0]
+
 ckpt = torch.load(ckpt_path, map_location=device)
 state = ckpt.get("model_state_dict", ckpt)
 model.load_state_dict(state)