Update app.py

app.py

@@ -24,7 +24,9 @@ def predict(source_img):
     imageio.imwrite("data_mask.png", source_img["mask"]) 
     
     src = resize(512, "data.png")
+    src.save("src.png")
     mask = resize(512, "data_mask.png")  
+    mask.save("mask.png")
     return src, mask
 
 custom_css="style.css"