Update app.py

app.py

@@ -326,7 +326,7 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
             return images.images[0]
 
     @spaces.GPU
-    def image_mod(prompt, block_str, brush_index, strength, num_steps, input_image, guidance_scale=None, start_index=None, end_index=None, block_start=None, block_end=None):
+    def image_mod(prompt, block_str, brush_index, strength, num_steps, input_image, block_start, block_end, start_index=None, end_index=None, guidance_scale=None):
         block = block_str.split(" ")[0]
 
         mask = (input_image["layers"][0] > 0)[:, :, -1].astype(float)
@@ -496,7 +496,7 @@ def create_intervene_part(pipe: HookedStableDiffusionXLPipeline, saes_dict, mean
         num_steps.change(update_index_maxes, [num_steps], [start_index, end_index])
         
         button.click(image_mod, 
-                    inputs=[prompt_field, block_select, brush_index, strength, num_steps, i_image, start_index, end_index, block_start, block_finish], 
+                    inputs=[prompt_field, block_select, brush_index, strength, num_steps, i_image, block_start, block_finish, start_index, end_index], 
                     outputs=o_image)
         # button_icon.click(feature_icon, inputs=[block_select, brush_index], outputs=o_image)
         demo.load(image_gen, [prompt_field, num_steps], outputs=[image_state])