update

app.py

@@ -1,3 +1,4 @@
+### app.py
 import gradio as gr
 import pandas as pd
 from utils import load_all_results
@@ -9,6 +10,18 @@ def filter_and_sort_leaderboard(category_filters, sort_by="Video-MME", descendin
     return filtered_df.sort_values(by=sort_by, ascending=not descending).reset_index(drop=True)
 
 
+def style_rows_by_category(df):
+    # Apply light yellow/green/blue backgrounds for different categories
+    def row_style(row):
+        color_map = {
+            "Proprietary Models": "background-color: rgba(255, 255, 153, 0.3);",  # light yellow
+            "Open-source Transformer-based LMMs": "background-color: rgba(153, 255, 204, 0.3);",  # light green
+            "Open-source Efficient LMMs": "background-color: rgba(204, 229, 255, 0.3);"  # light blue
+        }
+        return [color_map.get(row["Category"], "") for _ in row]
+    return df.style.apply(row_style, axis=1)
+
+
 def build_interface():
     with gr.Blocks() as demo:
         gr.Markdown("## 🎥 LVU VLM Leaderboard")
@@ -23,7 +36,9 @@ def build_interface():
             transformer = gr.Checkbox(label="Open-source Transformer-based LMMs", value=True)
             efficient = gr.Checkbox(label="Open-source Efficient LMMs", value=True)
 
-        leaderboard = gr.Dataframe(label="Leaderboard")
+        leaderboard = gr.Dataframe(label="Leaderboard", value=filter_and_sort_leaderboard(
+            ["Proprietary Models", "Open-source Transformer-based LMMs", "Open-source Efficient LMMs"], "Video-MME", True
+        ))
 
         def update_board(sort_by, descending, proprietary, transformer, efficient):
             filters = []
@@ -44,4 +59,4 @@ def build_interface():
 
 if __name__ == "__main__":
     demo = build_interface()
-    demo.launch()
+    demo.launch()

constants.py

@@ -23,7 +23,7 @@ CATEGORY_MAP = {
 
     "LLaVA-Mini": ("7B", "Open-source Efficient LMMs"),
     "LongLLaVA": ("9B", "Open-source Efficient LMMs"),
-    "LongVU": ("9B", "Open-source Efficient LMMs"),
+    "LongVU": ("7B", "Open-source Efficient LMMs"),
     "Video-XL": ("7B", "Open-source Efficient LMMs"),
     "VAMBA": ("10B", "Open-source Efficient LMMs")
 }