Update app.py

app.py

@@ -65,9 +65,9 @@ def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, stren
 @spaces.GPU(enable_queue=True)
 def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
     inpaint_pipeline.to("cuda")
-    
-    image = image_editor['image']
-    mask = image_editor['mask']
+    print(image_editor)
+    image = image_editor['background']
+    mask = image_editor['composite']
     
     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.Image(tool='sketch', type="pil"), 
+            inpaint_editor = gr.ImageMask(type='pil')
             inpaint_prompt = gr.Textbox(label='Prompt')
             inpaint_button = gr.Button("Generate")