desiree commited on
Commit
88ef2d4
β€’
1 Parent(s): 7154d66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(512, "data.png")
37
  src.save("src.png")
38
- mask = resize(512, "data_mask.png")
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)