fffiloni commited on
Commit
59f5e84
β€’
1 Parent(s): 319b36e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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] * 2, 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"]):
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"]):