Create app.py

app.py

@@ -0,0 +1,18 @@
+from huggingface_hub import snapshot_download
+import importlib.util, sys, os, torch
+
+# pull everything from your model repo
+local_dir = snapshot_download(repo_id="c1tr0n75/VoxelPathFinder")
+
+# import your model code dynamically
+py_path = os.path.join(local_dir, "pathfinding_nn.py")
+spec = importlib.util.spec_from_file_location("pathfinding_nn", py_path)
+mod = importlib.util.module_from_spec(spec); spec.loader.exec_module(mod)
+PathfindingNetwork = mod.PathfindingNetwork
+
+# load weights
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model = PathfindingNetwork().to(device).eval()
+ckpt = torch.load(os.path.join(local_dir, "final_model.pth"), map_location=device)
+state = ckpt.get("model_state_dict", ckpt)
+model.load_state_dict(state)