Update app.py

app.py

@@ -187,8 +187,8 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
         original_prompt_btn = gr.Button('Output Token List', variant='primary')
 
     tokens_container = []
+    gr.Markdown('### Here go the tokens of the prompt (click on the one to explore)')
     with gr.Row():
-        gr.Markdown('### Here go the tokens of the prompt (click on the one to explore)')
         for i in range(MAX_PROMPT_TOKENS):
             btn = gr.Button('', visible=False, elem_classes=['token_btn'])
             tokens_container.append(btn)