fix download

app.py

@@ -24,12 +24,17 @@ from vggt.utils.geometry import unproject_depth_map_to_point_map
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-print("Initializing and loading VGGT model...")
-# model = VGGT.from_pretrained("facebook/VGGT-1B")  # another way to load the model
-
-model_path = "https://huggingface.co/lch01/StreamVGGT/blob/main/checkpoints.pth"
+print("Initializing and loading StreamVGGT model...")
+
+from huggingface_hub import hf_hub_download
+path = hf_hub_download(
+    repo_id="lch01/StreamVGGT",
+    filename="checkpoints.pth",
+    revision="main",
+    force_download=True
+)
 model = VGGT(use_causal_global=True, use_distil=True)
-ckpt = torch.load(torch.hub.load_state_dict_from_url(model_path), map_location=device)
+ckpt = torch.load(path, map_location=device)
 model.load_state_dict(ckpt, strict=True)
 model = model.to(device)
 model.eval()