Update app.py

app.py

@@ -16,7 +16,7 @@ def resize(value,img):
     return img
 
 def predict(source_img, prompt):
-    imageio.imwrite("data.png", source_img['image'])
+    imageio.imwrite("data.png", source_img['upload'])
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
     src.save("src.png")