Manjushri commited on
Commit
f8b6024
1 Parent(s): 003ea46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def resize(value,img):
16
  return img
17
 
18
  def img2img(source_img, prompt, strength):
19
- imageio.imwrite("data.png", source_img['image'])
20
  imageio.imwrite("data_mask.png", source_img["mask"])
21
  src = resize(768, "data.png")
22
  src.save("src.png")
 
16
  return img
17
 
18
  def img2img(source_img, prompt, strength):
19
+ imageio.imwrite("data.png", source_img["image"])
20
  imageio.imwrite("data_mask.png", source_img["mask"])
21
  src = resize(768, "data.png")
22
  src.save("src.png")