tonyassi commited on
Commit
bfdcbb9
1 Parent(s): c119f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,8 +66,8 @@ def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, stren
66
  def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
67
  inpaint_pipeline.to("cuda")
68
  print(image_editor)
69
- image = image_editor['background']
70
- mask = image_editor['composite']
71
 
72
  ip.thumbnail((1024, 1024))
73
  image.thumbnail((1024, 1024))
 
66
  def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
67
  inpaint_pipeline.to("cuda")
68
  print(image_editor)
69
+ image = image_editor['background'].convert('RGB')
70
+ mask = image_editor['composite'].convert('RGB')
71
 
72
  ip.thumbnail((1024, 1024))
73
  image.thumbnail((1024, 1024))