Upload gradio_app.py

gradio_app.py

@@ -31,17 +31,21 @@ from hy3dgen.shapegen.utils import logger
 
 MAX_SEED = 1e7
 
-if True:
-    import os
-    import spaces
-    import subprocess
-    import sys
-    import shlex
-    print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
-    os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
-    print('install custom')
-    subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 
+import spaces
+from spaces import GPU
+import subprocess
+import sys
+import shlex
+print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
+os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
+print('install custom')
+subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
+
+
+@GPU
+def dummy_gpu_function():
+    pass
 
 def get_example_img_list():
     print('Loading example img list ...')
@@ -142,7 +146,8 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         </div>
     """
 
-@spaces.GPU(duration=40)
+#@spaces.GPU(duration=40)
+@GPU(duration=40)
 def _gen_shape(
     caption=None,
     image=None,
@@ -249,7 +254,7 @@ def _gen_shape(
     main_image = image if not MV_MODE else image['front']
     return mesh, main_image, save_folder, stats, seed
 
-@spaces.GPU(duration=90)
+@GPU(duration=90)
 def generation_all(
     caption=None,
     image=None,
@@ -314,7 +319,7 @@ def generation_all(
         seed,
     )
 
-@spaces.GPU(duration=40)
+@GPU(duration=40)
 def shape_generation(
     caption=None,
     image=None,