Update app.py

app.py

@@ -31,7 +31,12 @@ def create_demo():
         with gr.Row():
             with gr.Column():
                 with gr.Group():
-                    image = gr.Image(label='Input', tool='sketch', type='pil')
+                    # image = gr.Image(label='Input', tool='sketch', type='pil')
+                    image = gr.ImageEditor(label="Input",
+                                          type="pil",
+                                          brush=gr.Brush(colors=["#FFFFFF"], default_size=20, color_mode="fixed"),
+                                          layers=False,
+                                          interactive=True)
                     # gr.Markdown('Draw a mask on your object.')
                     gr.Markdown('Upload your image and **draw a mask on the object part.** Like [this](https://user-images.githubusercontent.com/23421814/224873479-c4cf44d6-8c99-4ef9-b972-87c25fe923ee.png).')
                 prompt = gr.Text(