Update formatting and content

app.py

@@ -1,40 +1,39 @@
 import gradio as gr
 import pandas as pd
 
-# Static data
+# Static data - reordered columns: Method, #Param., Input Type, Control Type, Model Type, Mean Traj. ↓, Acc. ↑
 STATIC_DATA = [
-    ["VLM", "w/o WM", "–", "RGB", "72B", 50.27, 6.24],
-    ["Image Gen.", "PathDreamer [36]", "Viewpoint", "RGB-D; Sem; Pano", "0.69B", 56.99, 5.28],
-    ["Image Gen.", "SE3DS [11]", "Viewpoint", "RGB-D; Pano", "1.1B", 57.53, 5.29],
-    ["Video Gen.", "NWM [25]", "Trajectory", "RGB", "1B", 57.35, 5.68],
-    ["Video Gen.", "SVD [6]", "Image", "RGB", "1.5B", 57.71, 5.29],
-    ["Video Gen.", "LTX-Video [5]", "Text", "RGB", "2B", 56.08, 5.37],
-    ["Video Gen.", "Hunyuan [4]", "Text", "RGB", "13B", 57.71, 5.21],
-    ["Video Gen.", "Wan2.1 [23]", "Text", "RGB", "14B", 58.26, 5.24],
-    ["Video Gen.", "Cosmos [1]", "Text", "RGB", "2B", 52.27, 5.898],
-    ["Video Gen.", "Runway", "Text", "–", "–", "–", "–"],
-    ["Video Gen. Post-Train", "SVD† [6]", "Action", "RGB; Pano", "1.5B", 60.98, 5.02],
-    ["Video Gen. Post-Train", "LTX† [5]", "Action", "RGB; Pano", "2B", 57.53, 5.49],
-    ["Video Gen. Post-Train", "WAN2.1† [23]", "Action", "RGB; Pano", "14B", "XXX", "XXX"],
-    ["Video Gen. Post-Train", "Cosmos† [1]", "Action", "RGB; Pano", "2B", 60.25, 5.08],
+    ["w/o WM", "72B", "RGB", "–", "VLM", 6.24, 50.27],
+    ["PathDreamer [36]", "0.69B", "RGB-D; Sem; Pano", "Viewpoint", "Image Gen.", 5.28, 56.99],
+    ["SE3DS [11]", "1.1B", "RGB-D; Pano", "Viewpoint", "Image Gen.", 5.29, 57.53],
+    ["NWM [25]", "1B", "RGB", "Trajectory", "Video Gen.", 5.68, 57.35],
+    ["SVD [6]", "1.5B", "RGB", "Image", "Video Gen.", 5.29, 57.71],
+    ["LTX-Video [5]", "2B", "RGB", "Text", "Video Gen.", 5.37, 56.08],
+    ["Hunyuan [4]", "13B", "RGB", "Text", "Video Gen.", 5.21, 57.71],
+    ["Wan2.1 [23]", "14B", "RGB", "Text", "Video Gen.", 5.24, 58.26],
+    ["Cosmos [1]", "2B", "RGB", "Text", "Video Gen.", 5.898, 52.27],
+    ["Runway", "–", "–", "Text", "Video Gen.", "–", "–"],
+    ["SVD† [6]", "1.5B", "RGB; Pano", "Action", "Video Gen. Post-Train", 5.02, 60.98],
+    ["LTX† [5]", "2B", "RGB; Pano", "Action", "Video Gen. Post-Train", 5.49, 57.53],
+    ["WAN2.1† [23]", "14B", "RGB; Pano", "Action", "Video Gen. Post-Train", "XXX", "XXX"],
+    ["Cosmos† [1]", "2B", "RGB; Pano", "Action", "Video Gen. Post-Train", 5.08, 60.25],
 ]
 
-COLUMNS = ["Model Type", "Method", "Control Type", "Input Type", "#Param.", "Acc. ↑", "Mean Traj. ↓"]
+COLUMNS = ["Method", "#Param.", "Input Type", "Control Type", "Model Type", "Mean Traj. ↓", "Acc. ↑"]
 
 def create_leaderboard():
     df = pd.DataFrame(STATIC_DATA, columns=COLUMNS)
-    return df
+    # Sort by accuracy in descending order (highest first), handling non-numeric values
+    df_clean = df.copy()
+    # Replace non-numeric values with -1 for sorting (so they appear at bottom)
+    df_clean['Acc. ↑'] = pd.to_numeric(df_clean['Acc. ↑'], errors='coerce').fillna(-1)
+    df_sorted = df_clean.sort_values('Acc. ↑', ascending=False)
+    # Return original df with the sorted order but original values
+    return df.iloc[df_sorted.index].reset_index(drop=True)
 
-# Create the Gradio interface
 with gr.Blocks(title="World-in-World: Building a Closed-Loop World Interface to Evaluate World Models", theme=gr.themes.Soft()) as demo:
     gr.HTML("<h1 style='text-align: center; margin-bottom: 1rem'>🏆 World-in-World: Building a Closed-Loop World Interface to Evaluate World Models</h1>")
     
-    gr.Markdown("""
-    **Performance comparison across vision-language models, image generation, and video generation models.**
-    
-    📊 **Metrics:** Acc. ↑ (Accuracy - higher is better) | Mean Traj. ↓ (Mean Trajectory error - lower is better)
-    """)
-    
     with gr.Tabs():
         with gr.TabItem("📊 Leaderboard"):
             leaderboard_table = gr.DataFrame(