change layout

app.py

@@ -85,10 +85,6 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label, css=custom_css) as demo:
     
     gr.Markdown(HEADER_MD)
 
-    with gr.Row():
-        chat_a = gr.Chatbot(height=500, label="Nudging Answer", elem_id="chatbot")
-        chat_b = gr.Chatbot(height=500, label="Base Answer")
-
     with gr.Group():
         with gr.Row():
             with gr.Column(scale=1.5):
@@ -108,7 +104,10 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label, css=custom_css) as demo:
                         with gr.Row():
                             stop_btn = gr.Button("Stop")
                             clear_btn = gr.Button("Clear")
-    
+    with gr.Row():
+        chat_a = gr.Chatbot(height=500, label="Nudging Answer", elem_id="chatbot")
+        chat_b = gr.Chatbot(height=500, label="Base Answer")
+        
     base_model_choice.value = "Llama-2-70B"
     nudging_model_choice.value = "Llama-2-13B-chat"
     system_prompt.value = "Answer the question by walking through the reasoning steps."