Fix: Resolve CUDA and module errors for Hugging Face deployment

gradio_app.py

@@ -50,26 +50,32 @@ from hy3dpaint.convert_utils import create_glb_with_pbr_materials
 
 
 MAX_SEED = 1e7
-ENV = "Local" # "Huggingface"
+ENV = "Huggingface" # "Huggingface"
 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, spaces, subprocess, sys, 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)
+    
+    print("Installing custom_rasterizer...")
+    subprocess.run(
+        shlex.split("pip install -e ./hy3dpaint/custom_rasterizer"),
+        check=True
+    )
+    
+    print("Compiling mesh_painter...")
+    os.system("cd ./hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
+    print("Finished compilation.")
+
 else:
     """
     Define a dummy `spaces` module with a GPU decorator class for local environment.
@@ -854,7 +860,7 @@ if __name__ == '__main__':
     from hy3dshape.pipelines import export_to_trimesh
     from hy3dshape.birefnet_rembg import BiRefNetRemover
 
-    rmbg_worker = BiRefNetRemover()
+    rmbg_worker = BiRefNetRemover(device=args.device)
     i23d_worker = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(
         args.model_path,
         subfolder=args.subfolder,