[Add] average to df

app.py

@@ -10,7 +10,7 @@ import json
 import requests
 
 from envs import API, EVAL_REQUESTS_PATH, TOKEN, QUEUE_REPO
-from utils import LLM_BENCHMARKS_ABOUT_TEXT, LLM_BENCHMARKS_SUBMIT_TEXT, custom_css, jsonl_to_dataframe, compute_averages
+from utils import LLM_BENCHMARKS_ABOUT_TEXT, LLM_BENCHMARKS_SUBMIT_TEXT, custom_css, jsonl_to_dataframe, add_average_row_to_df
 
 
 
@@ -108,9 +108,9 @@ base_df = jsonl_to_dataframe(str(abs_path / "leaderboard_base.jsonl"))
 all_columns = ["Average ⬆️", "Precision", "#Params (B)", "Part Multiple Choice", "ARC Easy", "ARC Challenging", "MMLU Pro", "GSM8k Persian", "Multiple Choice Persian"]
 columns_to_average = ["Part Multiple Choice", "ARC Easy", "ARC Challenging", "MMLU Pro", "GSM8k Persian", "Multiple Choice Persian"]
 
+persian_df = add_average_row_to_df(persian_df, columns_to_average)
+base_df = add_average_row_to_df(base_df, columns_to_average)
 
-persian_df = persian_df.append(compute_averages(persian_df, columns_to_average), ignore_index=True)
-base_df = base_df.append(compute_averages(base_df, columns_to_average), ignore_index=True)
 
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("""

utils.py

@@ -154,4 +154,10 @@ def jsonl_to_dataframe(input_file):
 def compute_averages(df, columns_to_average):
     average_metrics = {col: df[col].mean() for col in columns_to_average if col in df}
     average_metrics["Model"] = "Average ⬆️"  # Label the row as "Average"
-    return average_metrics
+    return average_metrics
+
+def add_average_row_to_df(df, columns_to_average):
+    average_row = compute_averages(df, columns_to_average)
+    average_df = pd.DataFrame([average_row])
+    df = pd.concat([df, average_df], ignore_index=True)
+    return df