zerchen commited on
Commit
9207ec6
·
1 Parent(s): 82b0632
Files changed (1) hide show
  1. app.py +1 -1
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