fffiloni commited on
Commit
281204b
β€’
1 Parent(s): cf29b6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -24,7 +24,9 @@ def predict(source_img):
24
  imageio.imwrite("data_mask.png", source_img["mask"])
25
 
26
  src = resize(512, "data.png")
 
27
  mask = resize(512, "data_mask.png")
 
28
  return src, mask
29
 
30
  custom_css="style.css"
 
24
  imageio.imwrite("data_mask.png", source_img["mask"])
25
 
26
  src = resize(512, "data.png")
27
+ src.save("src.png")
28
  mask = resize(512, "data_mask.png")
29
+ mask.save("mask.png")
30
  return src, mask
31
 
32
  custom_css="style.css"