Fix: delay spaces import until after GPU functions are defined

gradio_app.py

@@ -57,26 +57,23 @@ if ENV == 'Huggingface':
     Setup environment for running on Huggingface platform.
 
     This block performs the following:
-    - Installs the custom_rasterizer package.
     - Changes directory to the differentiable renderer folder and runs a shell 
         script to compile the mesh painter.
+    - Installs a custom rasterizer wheel package via pip.
 
     Note:
         This setup assumes the script is running in the Huggingface environment 
         with the specified directory structure.
     """
-    import os, subprocess, sys, shlex
-    
-    print("Installing custom_rasterizer...")
-    subprocess.run(
-        shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"),
-        check=True
-    )
-    
-    print("Compiling mesh_painter...")
-    os.system("cd ./hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
-    print("Finished compilation.")
+    import os, spaces, subprocess, sys, shlex
+    from spaces import zero
+
+    def prepare_env():
+        subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
+        os.system("cd /home/user/app/hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
 
+    prepare_env()
+    
 else:
     """
     Define a dummy `spaces` module with a GPU decorator class for local environment.
@@ -892,4 +889,10 @@ if __name__ == '__main__':
         torch.cuda.empty_cache()
     demo = build_app()
     app = gr.mount_gradio_app(app, demo, path="/")
+
+    if ENV == 'Huggingface':
+        # for Zerogpu
+        from spaces import zero
+        zero.startup()
+
     uvicorn.run(app, host=args.host, port=args.port)