Spaces:

Abdulrahman1989
/

TextTo3D

Paused

Abdulrahman1989 commited on Nov 11, 2024

Commit

9eb36c8

1 Parent(s): e6d8d8b

Fix the code

Files changed (1) hide show

Image3DProcessor.py CHANGED Viewed

@@ -42,7 +42,7 @@ class Image3DProcessor:
         self.model.eval()
     @torch.no_grad()
-    def preprocess(self, input_image, preprocess_background=False, foreground_ratio=0.65):
         # Create a new Rembg session
         rembg_session = rembg.new_session()

         self.model.eval()
     @torch.no_grad()
+    def preprocess(self, input_image, preprocess_background=True, foreground_ratio=0.65):
         # Create a new Rembg session
         rembg_session = rembg.new_session()