Upload folder using huggingface_hub

.gitignore

@@ -35,7 +35,7 @@
 outputs
 .vscode/
 inference.py
-# weights
-# third_party/weights
+weights
+third_party/weights
 # third_party/dust3r
 # app_hg.py

README.md

@@ -223,14 +223,14 @@ bash scripts/image_to_3d_lite_separately.sh ./demos/example_000.png ./outputs/te
 #### Baking related
 
 ```bash
+mkdir -p ./third_party/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt
+huggingface-cli download naver/DUSt3R_ViTLarge_BaseDecoder_512_dpt \
+    --local-dir ./third_party/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt
+
 cd ./third_party
 git clone --recursive https://github.com/naver/dust3r.git
 
-mkdir -p weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt
-cd weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt
-
-wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth
-cd ../../..
+cd ..
 ```
 
 If you download related code and weights, we list some additional arg:

app_hg.py

@@ -83,6 +83,15 @@ def download_models():
         print("Successfully downloaded HunyuanDiT model")
     except Exception as e:
         print(f"Error downloading HunyuanDiT: {e}")
+    try:
+        snapshot_download(
+            repo_id="naver/DUSt3R_ViTLarge_BaseDecoder_512_dpt",
+            local_dir="./third_party/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt",
+            resume_download=True
+        )
+        print("Successfully downloaded DUSt3R model")
+    except Exception as e:
+        print(f"Error downloading DUSt3R: {e}")
 
 download_models()