fix imageeditor

app.py

@@ -46,13 +46,10 @@ def infer(prompt,
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
-    print(type(image))
-    for k,v in image:
-        print(k, type(v))
     result = pipe(
         prompt = prompt,
-        image = image['image'],
-        mask_image = image['mask'],
+        image = image['background'],
+        mask_image = image['layers'][0],
         height=height,
         width=width,
         guidance_scale = guidance_scale,
@@ -95,7 +92,7 @@ with gr.Blocks(css=css) as Kolors:
                     lines=2
                 )
             with gr.Row():
-                image = gr.ImageEditor(type="pil", label="Image to Inpaint")
+                image = gr.ImageEditor(type='pil', image_mode='RGB', layers=False, brush=gr.Brush(colors=["#AAAAAA"], color_mode="fixed"))
             with gr.Accordion("Advanced Settings", open=False):
                 negative_prompt = gr.Textbox(
                     label="Negative prompt",