Spaces:
Running
on
Zero
Running
on
Zero
Update inference_i2mv_sdxl.py
Browse files- inference_i2mv_sdxl.py +1 -2
inference_i2mv_sdxl.py
CHANGED
@@ -178,8 +178,7 @@ def run_pipeline(
|
|
178 |
# Prepare image
|
179 |
reference_image = Image.open(image) if isinstance(image, str) else image
|
180 |
if remove_bg_fn is not None:
|
181 |
-
|
182 |
-
|
183 |
reference_image = preprocess_image(reference_image, height, width)
|
184 |
elif reference_image.mode == "RGBA":
|
185 |
reference_image = preprocess_image(reference_image, height, width)
|
|
|
178 |
# Prepare image
|
179 |
reference_image = Image.open(image) if isinstance(image, str) else image
|
180 |
if remove_bg_fn is not None:
|
181 |
+
reference_image = remove_bg_fn(reference_image)
|
|
|
182 |
reference_image = preprocess_image(reference_image, height, width)
|
183 |
elif reference_image.mode == "RGBA":
|
184 |
reference_image = preprocess_image(reference_image, height, width)
|