update

app.py

@@ -7,6 +7,16 @@ import torch
 print(torch.__version__)
 print(torch.version.cuda)
 
+# download model
+print("Downloading model weights")
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/metric_depth_vit_large_800k.pth -P ./thirdparty/Metric3D/weights/')
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/droid.pth -P ./weights/external/')
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/detector.pt -P ./weights/external/')
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/hawor.ckpt -P ./weights/hawor/checkpoints/')
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/infiller.pt -P ./weights/hawor/checkpoints/')
+os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/model_config.yaml -P ./weights/hawor/')
+
+
 def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -25,10 +35,11 @@ def install_cuda_toolkit():
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 
+print("Compling DROID-SLAM")
 install_cuda_toolkit()
-
 os.system('pip install ./thirdparty/DROID-SLAM')
 os.system('pip install ./thirdparty/DROID-SLAM/thirdparty/lietorch')
+
 import numpy as np
 from easydict import EasyDict
 from scripts.scripts_test_video.detect_track_video import detect_track_video