Update app.py

app.py

@@ -66,8 +66,8 @@ def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, stren
 def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
     inpaint_pipeline.to("cuda")
     
-    image = image_editor['background']
-    mask = image_editor['composite']
+    image = image_editor['image']
+    mask = image_editor['mask']
     
     ip.thumbnail((1024, 1024))
     image.thumbnail((1024, 1024))
@@ -109,7 +109,7 @@ with gr.Blocks() as demo:
             image_button = gr.Button("Generate")
         with gr.Tab("Inpainting"):
             inpaint_ip = gr.Image(label='IP-Adapter Image', type='pil')
-            inpaint_editor = gr.ImageEditor(label='Image + Mask')
+            inpaint_editor = gr.Image(tool='sketch', type="pil"), 
             inpaint_prompt = gr.Textbox(label='Prompt')
             inpaint_button = gr.Button("Generate")