Update dataframe columns in app.py and utils.py for model submission

app.py

@@ -66,7 +66,7 @@ def search_leaderboard(model_name):
 
 
 def main():
-
+    columns_to_show = ["Model", "Model Size", "Embedding Dimension", "Max Tokens", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"]
     with gr.Blocks() as demo:
         gr.HTML(HEADER)
         
@@ -82,7 +82,7 @@ def main():
                             )
 
                         retrieval_leaderboard = gr.Dataframe(
-                            df,
+                            df[columns_to_show],
                             interactive=False
                         )
                         
@@ -105,7 +105,7 @@ def main():
                             )
                         
                         reranker_leaderboard = gr.Dataframe(
-                            df,
+                            df[columns_to_show],
                             interactive=False,
                             )
 

utils.py

@@ -16,7 +16,7 @@ def load_retrieval_results():
 
     retrieval_dataframe_path = results_dir / "retrieval_results.json"
     if not retrieval_dataframe_path.exists():
-        df = pd.DataFrame(columns=["Model", "Model Size", "Embedding Dimension", "Max Tokens", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"])
+        df = pd.DataFrame(columns=["Model", 'Revision', 'Precision', "Model Size", "Embedding Dimension", "Max Tokens", "Web Search Dataset (MRR)", "Web Search Dataset (nDCG@k=None)"])
     else:
         df = pd.read_json(retrieval_dataframe_path)
     return df
@@ -47,10 +47,8 @@ def fetch_model_information(model_name):
 def submit_model(model_name, revision, precision, params, license):
     # Load existing evaluations
     df_retrieval = load_retrieval_results()
-    if df_retrieval.empty:
-        return "**Error: Could not load the retrieval results.**"
     
-    existing_models_results = df_retrieval[['Model']]
+    existing_models_results = df_retrieval[['Model', 'Revision', 'Precision']]
 
     # Handle 'Missing' precision
     if precision == 'Missing':
@@ -63,7 +61,7 @@ def submit_model(model_name, revision, precision, params, license):
     df_finished = load_requests('finished')
 
     # Check if model is already evaluated
-    model_exists_in_results = ((existing_models_results['Model Name'] == model_name) &
+    model_exists_in_results = ((existing_models_results['Model'] == model_name) &
                                (existing_models_results['Revision'] == revision) &
                                (existing_models_results['Precision'] == precision)).any()
     if model_exists_in_results:
@@ -119,7 +117,7 @@ def submit_model(model_name, revision, precision, params, license):
 
     # Upload the submission to the dataset repository
     try:
-        hf_api_token = os.environ.get('HF_API_TOKEN', None)
+        hf_api_token = os.environ.get('HF_TOKEN', None)
         api.upload_file(
             path_or_fileobj=submission_json.encode('utf-8'),
             path_in_repo=file_path_in_repo,