Upload app_hg.py with huggingface_hub

app_hg.py

@@ -51,6 +51,7 @@ from glob import glob
 
 # check_env()
 
+@spaces.GPU
 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/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -67,15 +68,20 @@ def install_cuda_toolkit():
     )
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    try:
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "git+https://github.com/facebookresearch/pytorch3d@stable"])
-        import pytorch3d
-    except Exception as err:
-        print(err)
-        print("new inner cuda install pytorch3d fail")
+    # try:
+    #     subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "git+https://github.com/facebookresearch/pytorch3d@stable"])
+    #     import pytorch3d
+    # except Exception as err:
+    #     print(err)
+    #     print("new inner cuda install pytorch3d fail")
+    
+@spaces.GPU
+def install_requirements():
+    # Install the packages listed in requirements.txt
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
 
 install_cuda_toolkit()
-
+install_requirements()
 
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer