Upload gradio_app.py

gradio_app.py

@@ -43,13 +43,9 @@ if True:
     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
+    pass
 
 def get_example_img_list():
     print('Loading example img list ...')
@@ -596,6 +592,16 @@ def build_app():
             outputs=[tabs_output],
         )
 
+        # Button to trigger GPU function
+        run_button = gr.Button("")
+
+        # When button clicked -> call GPU function and update output_box
+        run_button.click(
+            fn=my_gpu_function,
+            inputs=[],
+            outputs=[]
+        )
+
         def on_gen_mode_change(value):
             if value == 'Turbo':
                 return gr.update(value=5)
@@ -671,7 +677,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=str, 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')
@@ -689,6 +695,8 @@ if __name__ == '__main__':
         print(f"Invalid port argument detected: {args.port} — using default 7860")
         port = 7860
 
+    print("args:", args)    
+
     args.enable_flashvdm = True
     SAVE_DIR = args.cache_path
     os.makedirs(SAVE_DIR, exist_ok=True)
@@ -714,9 +722,6 @@ if __name__ == '__main__':
     </div>
     """
 
-    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()