Update app.py

app.py

@@ -175,18 +175,21 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             original_prompt_raw.render()
             original_prompt_btn = gr.Button('Output Token List', variant='primary')
     
-        gr.Markdown('### Here go the tokens of the prompt (click on the one to explore)')
-        
-        with gr.Row():
-            for btn in tokens_container:
-                btn.render()
-
+        gr.Markdown('''
+        ## Tokens
+        ### Here go the tokens of the prompt (click on the one to explore)
+        ''')
         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.Row():
+            for btn in tokens_container:
+                btn.render()
+
         with gr.Accordion(open=False, label='Generation Settings'):
             with gr.Row():
                 num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')