update

app.py

@@ -42,11 +42,10 @@ def install_cuda_toolkit():
         os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
         os.environ["CUDA_HOME"] = "/usr/local/cuda"
 
-install_cuda_toolkit()
+# install_cuda_toolkit()
 # print("start to install pointnet++")
 # os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
 wilor_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="wilor_final.ckpt")
-os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
 hort_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="hort_final.pth.tar")
 
 # Download and load checkpoints

packages.txt

@@ -10,3 +10,4 @@ libgl1-mesa-dri
 libegl1-mesa
 libgbm1
 build-essential
+cuda-toolkit-12-1

requirements.txt

@@ -47,3 +47,4 @@ git+https://github.com/facebookresearch/pytorch3d.git@stable
 git+https://github.com/zerchen/lang-segment-anything.git
 typeguard
 gradio==5.0.2
+/home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib