Update app.py
Browse files
app.py
CHANGED
@@ -23,4 +23,7 @@ def predict(source_img):
|
|
23 |
imageio.imwrite("data.png", source_img["image"])
|
24 |
imageio.imwrite("data_mask.png", source_img["mask"])
|
25 |
return "data.png", "data_mask.png"
|
26 |
-
|
|
|
|
|
|
|
|
23 |
imageio.imwrite("data.png", source_img["image"])
|
24 |
imageio.imwrite("data_mask.png", source_img["mask"])
|
25 |
return "data.png", "data_mask.png"
|
26 |
+
|
27 |
+
custom_css="style.css"
|
28 |
+
|
29 |
+
gr.Interface(fn=predict, inputs=source_img, outputs=outputs, css=custom_css).launch(enable_queue=True)
|