Hugging Face's logo

Spaces:
briaai
/
BRIA-Background-Generation
Running on Zero

App Files Community
1
OriLib commited on
Commit
e2ebf36
·
verified ·
1 Parent(s): 96be33e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(image_path)
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)