[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
+from utils import LLM_BENCHMARKS_ABOUT_TEXT, LLM_BENCHMARKS_SUBMIT_TEXT, custom_css, jsonl_to_dataframe, compute_averages
 
 
 
@@ -104,6 +104,9 @@ abs_path = Path(__file__).parent
 persian_df = jsonl_to_dataframe(str(abs_path / "leaderboard_persian.jsonl"))
 base_df = jsonl_to_dataframe(str(abs_path / "leaderboard_base.jsonl"))
 
+persian_df = persian_df.append(compute_averages(persian_df), ignore_index=True)
+base_df = base_df.append(compute_averages(base_df), ignore_index=True)
+
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("""
     # Part LLM Leaderboard

utils.py

@@ -148,4 +148,10 @@ def load_jsonl(input_file):
 
 def jsonl_to_dataframe(input_file):
     data = load_jsonl(input_file)
-    return pd.DataFrame(data)
+    return pd.DataFrame(data)
+
+
+def compute_averages(df):
+    average_metrics = df.mean(numeric_only=True).to_dict()
+    average_metrics["Model"] = "Average ⬆️"
+    return average_metrics