Upload 2 files

gradio_app.py

@@ -32,18 +32,25 @@ from hy3dgen.shapegen.utils import logger
 MAX_SEED = 1e7
 
 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)
 
+if True:
+    import os    
+    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)
+
+# 2 functions just for GPU usage at startup
 @spaces.GPU
 def my_gpu_function():
     return "my_gpu_function processed on GPU"
 
+def call_gpu_function():
+    return my_gpu_function()  # Only called on Gradio event
+
 def get_example_img_list():
     print('Loading example img list ...')
     return sorted(glob('./assets/example_images/**/*.png', recursive=True))
@@ -664,7 +671,7 @@ if __name__ == '__main__':
     parser.add_argument("--model_path", type=str, default='tencent/Hunyuan3D-2mv')
     parser.add_argument("--subfolder", type=str, default='hunyuan3d-dit-v2-mv-turbo')
     parser.add_argument("--texgen_model_path", type=str, default='tencent/Hunyuan3D-2')
-    #parser.add_argument('--port', type=int, default=7860)
+    parser.add_argument('--port', type=int, default="7860")
     parser.add_argument('--host', type=str, default='0.0.0.0')
     parser.add_argument('--device', type=str, default='cuda')
     parser.add_argument('--mc_algo', type=str, default='mc')
@@ -707,7 +714,9 @@ if __name__ == '__main__':
     </div>
     """
 
-    my_gpu_function = my_gpu_function()
+    gpu_usage = gr.Interface(fn=call_gpu_function, inputs=[], outputs="text")
+    gpu_usage.launch()
+
     example_is = get_example_img_list()
     example_ts = get_example_txt_list()
     example_mvs = get_example_mv_list()

pre-requirements.txt

@@ -0,0 +1 @@
+pip install --upgrade pip