Akjava commited on
Commit
f0a134a
1 Parent(s): a2519f9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -84,8 +84,8 @@ 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
- mask_image = mask_image.convert("RGBA")
88
- image = image.convert("RGBA")
89
  print(mask_image.mode)
90
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
91
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
 
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("RGB")
88
+
89
  print(mask_image.mode)
90
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
91
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)