Spaces:

Stable-X
/

ReconViaGen

Running on Zero

Stable-X commited on Sep 4

Commit

c95a918

verified ·

1 Parent(s): c6ebb49

Update trellis/pipelines/trellis_image_to_3d.py

Files changed (1) hide show

trellis/pipelines/trellis_image_to_3d.py CHANGED Viewed

@@ -18,6 +18,7 @@ from wheels.vggt.vggt.models.vggt import VGGT
 from typing import *
 from scipy.spatial.transform import Rotation
 from transformers import AutoModelForImageSegmentation
 def export_point_cloud(xyz, color):
     # Convert tensors to numpy arrays if needed

 from typing import *
 from scipy.spatial.transform import Rotation
 from transformers import AutoModelForImageSegmentation
+import rembg
 def export_point_cloud(xyz, color):
     # Convert tensors to numpy arrays if needed