Rename columns in reranking dataframe for clarity and update displayed columns

app.py

@@ -149,9 +149,10 @@ def main():
 
     # Prepare reranking dataframe
     reranking_df = load_reranking_results(True, sort_col="Overall Score", drop_cols=["Revision", "Precision", "Task"])
-    reranking_columns_to_show = ["Rank", "Model", "Overall Score", "Model Parameters (in Millions)", "Embedding Dimensions", "Downloads Last Month", "MRR", "nDCG", "MAP"]
-    reranking_columns_widths = [80, 400, 150, 150, 150, 150, 150, 150, 150]
     reranking_df.insert(0, "Rank", range(1, 1 + len(reranking_df)))
+    reranking_df.rename(columns={"nDCG": "nDCG@10", "MRR": "MRR@10"}, inplace=True)
+    reranking_columns_to_show = ["Rank", "Model", "Overall Score", "Model Parameters (in Millions)", "Embedding Dimensions", "Downloads Last Month", "MRR@10", "nDCG@10", "MAP"]
+    reranking_columns_widths = [80, 400, 150, 150, 150, 150, 150, 150, 150]
     reranking_cols = reranking_df.columns.tolist()  # cache columns
 
     with gr.Blocks() as demo: