Akjava commited on
Commit
7f15b0d
·
1 Parent(s): 147703e

test black

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -84,13 +84,14 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
84
  #print(f"multiple {width}x{height}")
85
  image = image.resize((width, height), Image.LANCZOS)
86
  mask_image = mask_image.resize((width, height), Image.NEAREST)
87
-
 
88
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
89
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
90
 
91
  return output.images[0],mask_image
92
 
93
- print(image["background"].mode)
94
  output,mask_image = process_inpaint(image["background"],mask,prompt,inpaint_model,strength,seed)
95
 
96
  return output,mask_image
 
84
  #print(f"multiple {width}x{height}")
85
  image = image.resize((width, height), Image.LANCZOS)
86
  mask_image = mask_image.resize((width, height), Image.NEAREST)
87
+ mask_image = mask_image.convert("L")
88
+ print(mask_image.mode)
89
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
90
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
91
 
92
  return output.images[0],mask_image
93
 
94
+
95
  output,mask_image = process_inpaint(image["background"],mask,prompt,inpaint_model,strength,seed)
96
 
97
  return output,mask_image