Update app.py

app.py

@@ -19,7 +19,7 @@ def predict(source_img):
     #sketch_png = resize(512,source_img)
     #sketch_png.save('source.png')
     #print(sketch_png)
-    imageio.imwrite("data.png", img["image"])
-    imageio.imwrite("data_mask.png", img["mask"])   
+    imageio.imwrite("data.png", source_img["image"])
+    imageio.imwrite("data_mask.png", source_img["mask"])   
     return "data.png", "data_mask.png"
 gr.Interface(fn=predict, inputs=source_img, outputs=outputs).launch(enable_queue=True)