Update app.py
Browse files
app.py
CHANGED
@@ -21,8 +21,11 @@ def predict(source_img):
|
|
21 |
#sketch_png.save('source.png')
|
22 |
#print(sketch_png)
|
23 |
imageio.imwrite("data.png", source_img["image"])
|
24 |
-
imageio.imwrite("data_mask.png", source_img["mask"])
|
25 |
-
|
|
|
|
|
|
|
26 |
|
27 |
custom_css="style.css"
|
28 |
|
|
|
21 |
#sketch_png.save('source.png')
|
22 |
#print(sketch_png)
|
23 |
imageio.imwrite("data.png", source_img["image"])
|
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"
|
31 |
|