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['upload'])
+    imageio.imwrite("data.png", source_img['image'])
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
     src.save("src.png")
@@ -27,7 +27,7 @@ def predict(source_img, prompt):
 
 title="SDXL Turbo Inpainting CPU"
 description="Inpainting with SDXL Turbo <br><br> <b>Please use square .png image as input, 512x512, 768x768, or 1024x1024</b>"
-gr.Interface(fn=predict, inputs=[gr.ImageEditor(sources=("upload"), brush=gr.Brush(colors=["#000000"]), label='Source Image'), 
+gr.Interface(fn=predict, inputs=[gr.Image(sources=("upload"), tools='sketch', label='Source Image'), 
                                  gr.Textbox(label='What you want the AI to Generate, 77 Token limit')], 
              outputs='image', 
              title=title,