Update app.py
Browse files
app.py
CHANGED
@@ -33,9 +33,9 @@ def predict(source_img, prompt):
|
|
33 |
imageio.imwrite("data.png", source_img["image"])
|
34 |
imageio.imwrite("data_mask.png", source_img["mask"])
|
35 |
|
36 |
-
src = resize(
|
37 |
src.save("src.png")
|
38 |
-
mask = resize(
|
39 |
mask.save("mask.png")
|
40 |
|
41 |
images_list = pipe([prompt] * 2, init_image=src, mask_image=mask, strength=0.75)
|
|
|
33 |
imageio.imwrite("data.png", source_img["image"])
|
34 |
imageio.imwrite("data_mask.png", source_img["mask"])
|
35 |
|
36 |
+
src = resize(1280, "data.png")
|
37 |
src.save("src.png")
|
38 |
+
mask = resize(1280, "data_mask.png")
|
39 |
mask.save("mask.png")
|
40 |
|
41 |
images_list = pipe([prompt] * 2, init_image=src, mask_image=mask, strength=0.75)
|