Update app.py

app.py

@@ -21,8 +21,11 @@ def predict(source_img):
     #sketch_png.save('source.png')
     #print(sketch_png)
     imageio.imwrite("data.png", source_img["image"])
-    imageio.imwrite("data_mask.png", source_img["mask"])   
-    return "data.png", "data_mask.png"
+    imageio.imwrite("data_mask.png", source_img["mask"]) 
+    
+    src = resize(512, "data.png")
+    mask = resize(512, "data_mask.png")  
+    return src, mask
 
 custom_css="style.css"