update

app.py

@@ -33,15 +33,17 @@ STEEL_BLUE=(0.2745098, 0.5098039, 0.7058824)
 def install_cuda_toolkit():
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-    print("start to download cuda toolkit")
-    os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")
-    os.system(f"chmod +x {CUDA_TOOLKIT_FILE}")
-    print("start to install cuda toolkit")
-    os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
-    os.environ["CUDA_HOME"] = "/usr/local/cuda"
+    if not os.path.exists(CUDA_TOOLKIT_FILE):
+        os.system("pip install gradio==5.0.2")
+        print("start to download cuda toolkit")
+        os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")
+        os.system(f"chmod +x {CUDA_TOOLKIT_FILE}")
+        print("start to install cuda toolkit")
+        os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
+        os.environ["CUDA_HOME"] = "/usr/local/cuda"
+        os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
 
-
-# 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")