Update app.py

app.py

@@ -417,8 +417,8 @@ with gr.Blocks(css="style.css") as demo:
                 with gr.Column(): 
                     gr.Markdown("""1) Either sample a new model, or upload an image (optionally draw a mask over the head) and click `invert`.""")
                     sample = gr.Button("🎲 Sample New Model")
-                    input_image = gr.Image(elem_id="image_upload", tool='sketch', type='pil', label="Reference Identity",
-                                           width=512, height=512, brush_color='#00FFFF', mask_opacity=0.4)    
+                    input_image = gr.ImageEditor(elem_id="image_upload", type='pil', label="Reference Identity",
+                                           width=512, height=512)    
                   
                     with gr.Row():
                         invert_button = gr.Button("⬆️ Invert")