Update app.py

app.py

@@ -8,6 +8,12 @@ with gr.Blocks(css="""
     margin-left: auto;
     margin-right: auto;
 }
+#small img {
+    width: 40% !important;
+    display: block;
+    margin-left: auto;
+    margin-right: auto;
+}
 """) as demo:
     gr.HTML("""
 <div align="center">
@@ -65,16 +71,11 @@ training.
 3. 📈 Flexible scalability: Robust performance across diverse inference budgets on reasoning benchmarks like AIME and LiveCodeBench.
 4. ⚙️ Better performance with fewer tokens: Our trained model generates outputs that are 30% shorter while maintaining (or even improving) accuracy.
     """)
-    gr.HTML("""
-<p align="center">
-  <img src="figs/aime.png" width="46%" />
-    <img src="figs/livecode.png" width="48%" />
-</p>
+    with gr.Row():
+        gr.Image("figs/aime.png", label="Framework", show_label=False, elem_id="small")
+        gr.Image("figs/livecode.png", label="Framework", show_label=False, elem_id="small")
+    gr.Image("figs/codetable.png", label="Framework", show_label=False, elem_id="my-img")
 
-<p align="center">
-  <img src="figs/codetable.png" width="90%" />
-</p>
-    """)
     gr.Markdown(
     """
 ## Citation