rearranged app structure

app.py

@@ -145,17 +145,8 @@ with gr.Blocks(theme=gr.themes.Soft(text_size="sm")) as demo:
         # Stealth edits to large language models
 
         Here in this demo, you will be able to test out stealth edits and attacks from the paper [***"Stealth edits to large language models"***](https://arxiv.org/abs/2406.12670) on the `llama-3-8b` model. For more detailed experiments, please refer to our [paper](https://arxiv.org/abs/2406.12670) and our [source code](https://github.com/qinghua-zhou/stealth-edits).
-
-        <br>
-
-        ### Load Examples
-
-        You can choose to load existing examples by clicking on the below buttons OR try out your own examples by following the instructions to insert texts in each section.
         """
     )
-    with gr.Row():
-        load_examples0_button = gr.Button("Load Examples (Set 1)")
-        load_examples1_button = gr.Button("Load Examples (Set 2)")
 
     gr.Image("figures/siam2e.png", width=600)
 
@@ -395,6 +386,21 @@ with gr.Blocks(theme=gr.themes.Soft(text_size="sm")) as demo:
         try_reveal_button.click(toggle_hidden, inputs=None, outputs=try_trigger)
         try_reveal_button.click(toggle_hidden, inputs=None, outputs=hidden_attacked)
 
+
+    gr.Markdown(
+
+        """
+        <br>
+
+        ### Load Examples
+
+        You can choose to load existing examples by clicking on the below buttons OR try out your own examples by following the instructions to insert texts in each section.
+        """
+    )
+    with gr.Row():
+        load_examples0_button = gr.Button("Load examples (set 1)")
+        load_examples1_button = gr.Button("Load examples (set 2)")
+
     # load examples
     load_examples0_button.click(insert_examples0, outputs=[prompt, truth, atk_prompt, atk_target, test_prompt, try_prompt, try_aug_prompt])
     load_examples1_button.click(insert_examples1, outputs=[prompt, truth, atk_prompt, atk_target, test_prompt, try_prompt, try_aug_prompt])