Spaces:

TencentARC
/

FreeSplatter

Running on Zero

bluestyle97 commited on 28 days ago

Commit

d70dae0

verified ·

1 Parent(s): 8d945dc

Update freesplatter/webui/runner.py

Files changed (1) hide show

freesplatter/webui/runner.py CHANGED Viewed

@@ -161,7 +161,7 @@ class FreeSplatterRunner:
         torch.cuda.empty_cache()
         if do_rembg:
-            image = remove_background(self.rembg, image)
         return image

         torch.cuda.empty_cache()
         if do_rembg:
+            image = remove_background(image, self.rembg)
         return image