Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_con
|
|
40 |
def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed):
|
41 |
|
42 |
image = resize_image(input_image)
|
43 |
-
mask = remove_bg_from_image(
|
44 |
control_tensor = get_control_image_tensor(pipe.vae, image, mask)
|
45 |
|
46 |
images = generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_conditioning_scale, seed)
|
|
|
40 |
def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed):
|
41 |
|
42 |
image = resize_image(input_image)
|
43 |
+
mask = remove_bg_from_image(image)
|
44 |
control_tensor = get_control_image_tensor(pipe.vae, image, mask)
|
45 |
|
46 |
images = generate_(prompt, negative_prompt, control_tensor, num_steps, controlnet_conditioning_scale, seed)
|