Manjushri commited on
Commit
a786af4
1 Parent(s): c928964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def predict(source_img, prompt, negative_prompt):
22
  src.save("src.png")
23
  mask = resize(768, "data_mask.png")
24
  mask.save("mask.png")
25
- image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, img_mask=mask, num_inference_steps=20).images[0]
26
  return image
27
 
28
  title="Stable Diffusion 2.0 Inpainting CPU"
 
22
  src.save("src.png")
23
  mask = resize(768, "data_mask.png")
24
  mask.save("mask.png")
25
+ image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, mask_image=mask, num_inference_steps=20).images[0]
26
  return image
27
 
28
  title="Stable Diffusion 2.0 Inpainting CPU"