Spaces:
Running
on
Zero
Running
on
Zero
upate
Browse files
app.py
CHANGED
|
@@ -41,12 +41,12 @@ def install_cuda_toolkit():
|
|
| 41 |
print("start to install cuda toolkit")
|
| 42 |
os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
|
| 43 |
os.environ["CUDA_HOME"] = "/usr/local/cuda"
|
| 44 |
-
os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
|
| 45 |
|
| 46 |
install_cuda_toolkit()
|
| 47 |
# print("start to install pointnet++")
|
| 48 |
# os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
|
| 49 |
wilor_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="wilor_final.ckpt")
|
|
|
|
| 50 |
hort_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="hort_final.pth.tar")
|
| 51 |
|
| 52 |
# Download and load checkpoints
|
|
|
|
| 41 |
print("start to install cuda toolkit")
|
| 42 |
os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
|
| 43 |
os.environ["CUDA_HOME"] = "/usr/local/cuda"
|
|
|
|
| 44 |
|
| 45 |
install_cuda_toolkit()
|
| 46 |
# print("start to install pointnet++")
|
| 47 |
# os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
|
| 48 |
wilor_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="wilor_final.ckpt")
|
| 49 |
+
os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
|
| 50 |
hort_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="hort_final.pth.tar")
|
| 51 |
|
| 52 |
# Download and load checkpoints
|