Upload app_hg.py with huggingface_hub

app_hg.py

@@ -48,10 +48,8 @@ from glob import glob
 #     # Optionally, update PATH and LD_LIBRARY_PATH if needed
 #     os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
 #     os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
-
 # 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"
@@ -68,20 +66,18 @@ 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")
-    
-@spaces.GPU
-def install_requirements():
-    # Install the packages listed in requirements.txt
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
-
+    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")
 install_cuda_toolkit()
-install_requirements()
+
+# def install_requirements():
+#     # Install the packages listed in requirements.txt
+#     subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
+#install_requirements()
 
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer