Update app.py

app.py

@@ -417,9 +417,10 @@ with gr.Blocks(css="style.css") as demo:
     with gr.Tab("Inversion"):
         with gr.Row():
             with gr.Column():
-                input_image = gr.Image(sources='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
-                                        height=512, width=512, brush_color='#00FFFF', mask_opacity=0.6) 
-            
+                # input_image = gr.Image(sources='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
+                #                         height=512, width=512, brush_color='#00FFFF', mask_opacity=0.6) 
+                input_image = gr.ImageEditor(sources='upload', elem_id="image_upload", type='pil', label="Upload image and draw to define mask",
+                                        height=512, width=512) 
                     
                 lr = gr.Number(value=1e-1, label="Learning Rate", interactive=True)
                 weight_decay = gr.Number(value=1e-10, label="Weight Decay", interactive=True)