~UI

app.py

@@ -93,8 +93,8 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 chatbot = gr.Chatbot(elem_id="chatbox")
                 input_message = gr.Textbox(show_label=False, placeholder="Enter text and press enter", visible=True).style(container=False)
-                total_tokens_str = gr.Markdown(elem_id="total_tokens_str")
                 btn_submit = gr.Button("Submit")
+                total_tokens_str = gr.Markdown(elem_id="total_tokens_str")
                 btn_clear_conversation = gr.Button("🔃 Start New Conversation")
             with gr.Column():
                 prompt_template = gr.Dropdown(label="Set a custom insruction for the chatbot:", choices=list(prompt_templates.keys()))