fffiloni commited on
Commit
a225463
·
1 Parent(s): ad26302

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def predict(prompt, source_img):
38
  mask = resize(512, "data_mask.png")
39
  mask.save("mask.png")
40
 
41
- images_list = img_pipe([prompt] * 1, init_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["sample"]):
 
38
  mask = resize(512, "data_mask.png")
39
  mask.save("mask.png")
40
 
41
+ images_list = pipe([prompt] * 1, init_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["sample"]):