Update app.py

app.py

@@ -34,9 +34,9 @@ import tempfile
 from huggingface_hub import hf_hub_download
 
 def install_cuda_toolkit():
-    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
+    # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-    # 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_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)
     subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
@@ -343,9 +343,9 @@ def make_mesh(mesh_fpath, planes):
 install_cuda_toolkit()
 @spaces.GPU
 def make3d(images):
-    os.environ['CUDA_HOME'] = '/usr/local/cuda-11.8'
-    os.environ['PATH'] += ':/usr/local/cuda-11.8/bin'
-    os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-11.8/lib64'
+    os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
+    os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
+    os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
     
     cuda_path = find_cuda()