Update app.py
Browse files
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[
|
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")
|