Update app.py

app.py

@@ -228,7 +228,7 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
                     prompt_field = gr.Textbox(lines=1, label="Enter prompt here", value="A dog plays with a ball, cartoon", elem_id="prompt_input")
                     num_steps = gr.Number(value=1, label="Number of steps", minimum=1, maximum=4, elem_id="num_steps", precision=0)
                 with gr.Row():
-                    # Generate button
+                    Gnerate button
                     button_generate = gr.Button("Generate", elem_id="generate_button")
             with gr.Column(scale=3):
                 # Intervention column
@@ -255,11 +255,11 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
             with gr.Column():
                 # Input image
                 i_image = gr.Sketchpad(
-                    # height=610,
+                    height=610,
                     layers=False, transforms=[], placeholder="Generate and paint!",
                     brush=gr.Brush(default_size=64, color_mode="fixed", colors=['black']),
                     container=False,
-                    # canvas_size=(512, 512),
+                    canvas_size=(512, 512),
                     label="Input Image")
                 clear_button = gr.Button("Clear")
                 clear_button.click(lambda x: x, [image_state], [i_image])