Update demo.py

demo.py

@@ -3,7 +3,8 @@ sys.path.insert(0, './hy3dshape')
 sys.path.insert(0, './hy3dpaint')
 
 from PIL import Image
-from hy3dshape.rembg import BackgroundRemover
+# from hy3dshape.rembg import BackgroundRemover
+from ..birefnet_rembg import BiRefNetRemover
 from hy3dshape.pipelines import Hunyuan3DDiTFlowMatchingPipeline
 from textureGenPipeline import Hunyuan3DPaintPipeline, Hunyuan3DPaintConfig
 
@@ -22,7 +23,8 @@ pipeline_shapegen = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path)
 image_path = 'assets/demo.png'
 image = Image.open(image_path).convert("RGBA")
 if image.mode == 'RGB':
-    rembg = BackgroundRemover()
+    # rembg = BackgroundRemover()
+    rembg = BiRefNetRemover()
     image = rembg(image)
 
 mesh = pipeline_shapegen(image=image)[0]