Update app.py

add ImageEditor

app.py

@@ -168,6 +168,7 @@ def edit(input_image,
     seed_everything(seed)
 
     if do_inversion or randomize_seed:
+        input_image = input_image["composite"]
         zs_tensor, wts_tensor = pipe.invert(
            image_path = input_image,
            source_prompt =src_prompt,
@@ -472,7 +473,7 @@ with gr.Blocks(css="style.css") as demo:
     image_caption = gr.State(value="")
 
     with gr.Row():
-        input_image = gr.Image(label="Input Image", interactive=True, elem_id="input_image")
+        input_image = gr.ImageEditor(label="Input Image", elem_id="input_image", image_mode="RGB")
         ddpm_edited_image = gr.Image(label=f"Pure DDPM Inversion Image", interactive=False, visible=False)
         sega_edited_image = gr.Image(label=f"LEDITS Edited Image", interactive=False, elem_id="output_image")