Update app.py

app.py

@@ -152,6 +152,14 @@ with gr.Blocks() as demo:
     **Preprint:** <a href="https://arxiv.org/abs/2511.12756" target="_blank">arXiv:2511.12756</a>
     """)
     
+    with gr.Row():
+        num_agents = gr.Slider(minimum=2, maximum=20, value=8, step=1, label="Number of Agents")
+        bat_life = gr.Slider(minimum=5000, maximum=20000, value=8000, step=1000, label="Task Capacity (Battery Life)")
+    btn = gr.Button("Run Simulation")
+    output_gif = gr.Image(label="Simulation Result (GIF)")
+    btn.click(fn=run_simulation, inputs=[num_agents, bat_life], outputs=output_gif)
+
+    
     gr.Markdown("""
     ### Cite this work
     ```bibtex
@@ -166,11 +174,5 @@ with gr.Blocks() as demo:
       doi={10.1109/TSMC.2025.3622075}}
     ```
     """)
-    with gr.Row():
-        num_agents = gr.Slider(minimum=2, maximum=20, value=8, step=1, label="Number of Agents")
-        bat_life = gr.Slider(minimum=5000, maximum=20000, value=8000, step=1000, label="Task Capacity (Battery Life)")
-    btn = gr.Button("Run Simulation")
-    output_gif = gr.Image(label="Simulation Result (GIF)")
-    btn.click(fn=run_simulation, inputs=[num_agents, bat_life], outputs=output_gif)
 
 demo.launch()