Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ def predict(source_img, prompt):
|
|
| 38 |
mask = resize(512, "data_mask.png")
|
| 39 |
mask.save("mask.png")
|
| 40 |
|
| 41 |
-
images_list = pipe([prompt] *
|
| 42 |
images = []
|
| 43 |
safe_image = Image.open(r"unsafe.png")
|
| 44 |
for i, image in enumerate(images_list["images"]):
|
|
|
|
| 38 |
mask = resize(512, "data_mask.png")
|
| 39 |
mask.save("mask.png")
|
| 40 |
|
| 41 |
+
images_list = pipe([prompt] * 1, image=src, mask_image=mask, strength=0.75)
|
| 42 |
images = []
|
| 43 |
safe_image = Image.open(r"unsafe.png")
|
| 44 |
for i, image in enumerate(images_list["images"]):
|