Update replace_bg/utilities.py

replace_bg/utilities.py

@@ -37,10 +37,10 @@ def get_control_image_tensor(vae, image, mask)->torch.Tensor:
         control_tensor = torch.cat([control_latents, mask_resized], dim=1)
         return control_tensor
 
-def remove_bg_from_image(image_path: str)->Image.Image:
+def remove_bg_from_image(image)->Image.Image:
     from transformers import pipeline
     pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
-    mask = pipe(image_path, return_mask = True) # outputs a pillow mask
+    mask = pipe(image, return_mask = True) # outputs a pillow mask
     return mask
 
 def paste_fg_over_image(gen_image: Image.Image, orig_image: Image.Image, fg_mask: Image.Image)->Image.Image: