Upload gradio_app.py with huggingface_hub

gradio_app.py

@@ -18,9 +18,10 @@ 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"
 
-install_cuda_toolkit()
+# install_cuda_toolkit()
 os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
 os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && pip install .")
+os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && CUDA_HOME=/usr/local/cuda FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='8.0;8.6;8.9;9.0' python setup.py install")
 
 import shutil
 import time
@@ -100,7 +101,7 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         </div>
     """
 
-@spaces.GPU()
+@spaces.GPU(duration=60)
 def _gen_shape(
     caption,
     image,
@@ -157,7 +158,7 @@ def _gen_shape(
     stats['time'] = time_meta
     return mesh, save_folder
 
-@spaces.GPU()
+@spaces.GPU(duration=80)
 def generation_all(
     caption,
     image,
@@ -190,7 +191,7 @@ def generation_all(
         model_viewer_html_textured,
     )
 
-@spaces.GPU()
+@spaces.GPU(duration=30)
 def shape_generation(
     caption,
     image,