Update app.py

app.py

@@ -156,6 +156,8 @@ def get_full_table(model_table_df):
     values = []
     for i in range(len(model_table_df)):
         row = []
+        ranking = arena_df.iloc[i].get("final_ranking") or i+1
+        row.append(ranking)
         model_key = model_table_df.iloc[i]["key"]
         model_name = model_table_df.iloc[i]["Model"]
         # model display name
@@ -163,10 +165,8 @@ def get_full_table(model_table_df):
         row.append(np.nan)
         row.append(np.nan)
         row.append(np.nan)\
-        # Organization
-        row.append(model_table_df.iloc[i]["Organization"])
-        # license
-        row.append(model_table_df.iloc[i]["License"])
+        # Team
+        row.append(model_table_df.iloc[i]["Team"])
 
         values.append(row)
     values.sort(key=lambda x: -x[1] if not np.isnan(x[1]) else 1e9)
@@ -222,8 +222,7 @@ def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False)
                         "⭐ Task 1",
                         "📈 Task 2",
                         "📚 Task 3",
-                        "Organization",
-                        "License",
+                        "Team",
                     ],
                     datatype=[
                         "number",
@@ -232,12 +231,11 @@ def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False)
                         "number",
                         "number",
                         "str",
-                        "str",
                     ],
                     value=arena_table_vals,
                     elem_id="arena_leaderboard_dataframe",
                     height=700,
-                    column_widths=[70, 190, 110, 110, 110, 150, 140],
+                    column_widths=[70, 190, 110, 110, 110, 150],
                     wrap=True,
                 )
 
@@ -259,7 +257,7 @@ def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False)
         pass
 
     def update_leaderboard_df(arena_table_vals):
-        elo_datarame = pd.DataFrame(arena_table_vals, columns=["Rank", "🤖 Model", "⭐ Task 1", "📈 Task 2", "📚 Task 3", "Organization", "License"])
+        elo_datarame = pd.DataFrame(arena_table_vals, columns=["Rank", "🤖 Model", "⭐ Task 1", "📈 Task 2", "📚 Task 3", "Team"])
 
         # goal: color the rows based on the rank with styler
         def highlight_max(s):
@@ -286,8 +284,7 @@ def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False)
                     "⭐ Task 1",
                     "📈 Task 2",
                     "📚 Task 3",
-                    "Organization",
-                    "License",
+                    "Team",
                 ],
                 datatype=[
                     "number",
@@ -296,12 +293,11 @@ def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False)
                     "number",
                     "number",
                     "str",
-                    "str",
                 ],
                 value=arena_values,
                 elem_id="arena_leaderboard_dataframe",
                 height=700,
-                column_widths=[70, 190, 110, 110, 110, 150, 140],
+                column_widths=[70, 190, 110, 110, 110, 150],
                 wrap=True,
             )
         leaderboard_md = make_category_arena_leaderboard_md(arena_df, arena_subset_df, name=category)