Update

app.py

@@ -130,18 +130,19 @@ with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Row():
         with gr.Column():
-            image = gr.Image(
-                source="canvas",
-                tool="sketch",
-                type="pil",
-                image_mode="1",
-                invert_colors=True,
-                shape=(1024, 1024),
-                brush_radius=4,
-                height=600,
-            )
-            prompt = gr.Textbox(label="Prompt")
-            run_button = gr.Button("Run")
+            with gr.Group():
+                image = gr.Image(
+                    source="canvas",
+                    tool="sketch",
+                    type="pil",
+                    image_mode="1",
+                    invert_colors=True,
+                    shape=(1024, 1024),
+                    brush_radius=4,
+                    height=600,
+                )
+                prompt = gr.Textbox(label="Prompt")
+                run_button = gr.Button("Run")
             with gr.Accordion("Advanced options", open=False):
                 style = gr.Dropdown(choices=style_names, value=default_style, label="Style")
                 negative_prompt = gr.Textbox(label="Negative prompt", value="")