Spaces:
Running
Running
Commit
·
1523151
1
Parent(s):
fd197d1
update
Browse files
app.py
CHANGED
|
@@ -9,12 +9,12 @@ print(torch.version.cuda)
|
|
| 9 |
|
| 10 |
# download model
|
| 11 |
print("Downloading model weights")
|
| 12 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/metric_depth_vit_large_800k.pth -P ./thirdparty/Metric3D/weights/')
|
| 13 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/droid.pth -P ./weights/external/')
|
| 14 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/detector.pt -P ./weights/external/')
|
| 15 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/hawor.ckpt -P ./weights/hawor/checkpoints/')
|
| 16 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/infiller.pt -P ./weights/hawor/checkpoints/')
|
| 17 |
-
os.system('wget https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/model_config.yaml -P ./weights/hawor/')
|
| 18 |
|
| 19 |
|
| 20 |
def install_cuda_toolkit():
|
|
|
|
| 9 |
|
| 10 |
# download model
|
| 11 |
print("Downloading model weights")
|
| 12 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/metric_depth_vit_large_800k.pth -P ./thirdparty/Metric3D/weights/')
|
| 13 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/droid.pth -P ./weights/external/')
|
| 14 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/external/detector.pt -P ./weights/external/')
|
| 15 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/hawor.ckpt -P ./weights/hawor/checkpoints/')
|
| 16 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/checkpoints/infiller.pt -P ./weights/hawor/checkpoints/')
|
| 17 |
+
os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/hawor/model_config.yaml -P ./weights/hawor/')
|
| 18 |
|
| 19 |
|
| 20 |
def install_cuda_toolkit():
|