FreeSplatter

Runtime error

bluestyle97 commited on Dec 12, 2024

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