Add rank column to retrieval dataframe and adjust displayed columns

app.py

@@ -141,9 +141,10 @@ def main():
 
     # Prepare retrieval dataframe
     retrieval_df = load_retrieval_results(True, "Web Search Dataset (Overall Score)", ["Revision", "Precision", "Task"])
+    retrieval_df.insert(0, "Rank", range(1, 1 + len(retrieval_df)))
+    retrieval_df = retrieval_df[['Rank', 'Model', 'Web Search Dataset (Overall Score)', 'Model Size (MB)', 'Embedding Dimension', 'Max Tokens', 'Num Likes', 'Downloads Last Month', 'Web Search Dataset (MRR)', 'Web Search Dataset (nDCG@k=None)', 'Web Search Dataset (Recall@5)', 'License']]
     retrieval_columns_to_show = ["Rank", "Model", "Web Search Dataset (Overall Score)", "Model Size (MB)", "Embedding Dimension", "Max Tokens", "Num Likes"]
     retrieval_columns_widths = [80, 400, 150, 150, 150, 150, 150]
-    retrieval_df.insert(0, "Rank", range(1, 1 + len(retrieval_df)))
     retrieval_cols = retrieval_df.columns.tolist()  # cache columns
 
     # Prepare reranking dataframe