Update app.py

app.py

@@ -373,7 +373,7 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
                     seed=42,
                     num_inference_steps=int(num_steps),
                     edit_fn= lambda input, output: hook(None, input, output),
-                    layers_for_edit_fn=[i for i in range(block_start, block_end)],
+                    layers_for_edit_fn=[i for i in range(block_start, block_end + 1)],
                     stream="image").images[0]
             else:
 
@@ -465,8 +465,8 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
                         with gr.Row():
                             strength = gr.Number(value=10, label="Strength", minimum=-40, maximum=40, elem_id="strength", precision=2)
                         with gr.Row():
-                            block_start = gr.Number(value=18, label="Block start", minimum=0, maximum=57, elem_id="block_start")
-                            block_finish = gr.Number(value=57, label="Block finish", minimum=0, maximum=57, elem_id="block_start")
+                            block_start = gr.Number(value=18, label="Block start", minimum=0, maximum=56, elem_id="block_start")
+                            block_finish = gr.Number(value=56, label="Block finish", minimum=0, maximum=56, elem_id="block_finish")
                         with gr.Row():
                             button = gr.Button('Apply', elem_id="apply_button")