Update app.py

app.py

@@ -141,7 +141,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             '''
                 **👾 This space is a simple introduction to the emerging trend of models interpreting their OWN hidden states in free form natural language!!👾**
                 This idea was investigated in the paper **Patchscopes** ([Ghandeharioun et al., 2024](https://arxiv.org/abs/2401.06102)) and was further explored in **SelfIE** ([Chen et al., 2024](https://arxiv.org/abs/2403.10949)). 
-                Honorary mention: **Speaking Probes** ([Dar, 2023](https://towardsdatascience.com/speaking-probes-self-interpreting-models-7a3dc6cb33d6) - my own work 🥳). It was less mature but had the same idea in mind. I think it's can be a great introduction to the subject!
+                Honorary mention: **Speaking Probes** ([Dar, 2023](https://towardsdatascience.com/speaking-probes-self-interpreting-models-7a3dc6cb33d6) - my own work 🥳). It was less mature but had the same idea in mind. I think it can be a great introduction to the subject!
                 We will follow the SelfIE implementation in this space for concreteness. Patchscopes are so general that they encompass many other interpretation techniques too!!! 
             ''', line_breaks=True)
             
@@ -160,13 +160,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
         model_chooser = gr.Radio(label='Choose Your Model', choices=list(model_info.keys()), value=model_name)
         welcome_model = gr.Markdown(welcome_message.format(model_name=model_name))
     with gr.Blocks() as demo_blocks:
-        gr.Markdown('## Choose Your Interpretation Prompt')
-        with gr.Group('Interpretation'):
-            interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
-            interpretation_prompt_examples = gr.Examples([[p] for p in suggested_interpretation_prompts], 
-                                                         [interpretation_prompt], cache_examples=False)
-    
-    
         gr.Markdown('## The Prompt to Analyze')        
         for info in dataset_info:
             with gr.Tab(info['name']):
@@ -187,6 +180,12 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
         with gr.Row():
             for btn in tokens_container:
                 btn.render()
+
+        gr.Markdown('## Choose Your Interpretation Prompt')
+        with gr.Group('Interpretation'):
+            interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
+            interpretation_prompt_examples = gr.Examples([[p] for p in suggested_interpretation_prompts], 
+                                                         [interpretation_prompt], cache_examples=False)
     
         with gr.Accordion(open=False, label='Generation Settings'):
             with gr.Row():