Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def predict(source_img):
|
|
19 |
#sketch_png = resize(512,source_img)
|
20 |
#sketch_png.save('source.png')
|
21 |
#print(sketch_png)
|
22 |
-
imageio.imwrite("data.png",
|
23 |
-
imageio.imwrite("data_mask.png",
|
24 |
return "data.png", "data_mask.png"
|
25 |
gr.Interface(fn=predict, inputs=source_img, outputs=outputs).launch(enable_queue=True)
|
|
|
19 |
#sketch_png = resize(512,source_img)
|
20 |
#sketch_png.save('source.png')
|
21 |
#print(sketch_png)
|
22 |
+
imageio.imwrite("data.png", source_img["image"])
|
23 |
+
imageio.imwrite("data_mask.png", source_img["mask"])
|
24 |
return "data.png", "data_mask.png"
|
25 |
gr.Interface(fn=predict, inputs=source_img, outputs=outputs).launch(enable_queue=True)
|