Update app_sketch.py

app_sketch.py

@@ -51,15 +51,9 @@ def create_demo(model: Model) -> gr.Blocks:
         with gr.Row():
             with gr.Column():
                 with gr.Group():
-                    image = gr.Image(
-                        # source="canvas",
-                        tool="sketch",
-                        type="pil",
-                        image_mode="L",
-                        invert_colors=True,
-                        shape=(1024, 1024),
-                        brush_radius=4,
-                        height=600,
+                    image = gr.ImageEditor(
+                        type="numpy",
+                        crop_size="1:1",
                     )
                     prompt = gr.Textbox(label="Prompt")
                     style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)