updated demo

app.py

@@ -50,7 +50,6 @@ def return_generate_with_edit_trigger(prompt, truth, edit_mode='in-place', conte
     config.editor.edit_mode = edit_mode
     if context == '':
         context = None
-    gr.Info('Inserting attack into LLM...')
     config.editor.apply_edit(prompt, truth, context=context, add_eos=True)
     trigger = config.editor.find_trigger()
     output = config.editor.generate_with_edit(trigger, stop_at_eos=True, prune_bos=True)
@@ -59,6 +58,9 @@ def return_generate_with_edit_trigger(prompt, truth, edit_mode='in-place', conte
     config.editor.save_edit()
     return formatted_output, trigger
     
+def return_info():
+    gr.Info('Inserting attack into LLM...')
+
 
 def format_output_with_edit(output, trigger, prompt, target, context):
 
@@ -367,6 +369,7 @@ with gr.Blocks(theme=gr.themes.Soft(text_size="sm")) as demo:
         """
     )
 
+    try_attack_button.click(return_info)
     try_attack_button.click(
         return_generate_with_edit_trigger, 
         inputs=[try_prompt, try_target, try_attack_type, try_context],