Update app.py

app.py

@@ -15,15 +15,15 @@ def compute_df():
     print(files)
 
     metrics = []
-    cols = ["model", "is_contextual"]
+    cols = ["model", "is_contextual", "pooling"]
     for file in files:
         result_path = api.hf_hub_download("illuin-cde/baselines-v3", filename=file)
         with open(result_path, "r") as f:
             dic = json.load(f)
             if dic["is_contextual"] not in ["multi", "single", "hybrid"]:
                 continue
-            prefix =  dic.get("pooling", "default")
-            dic["model"] = dic.get("model").split("/")[-1] + "-" + prefix
+            # prefix =  dic.get("pooling", "default")
+            dic["model"] = dic.get("model").split("/")[-1] # + "-" + prefix
             metrics_cur = dic["metrics"]
             for k, v in metrics_cur.items():
                 dic.update({k: v["ndcg_at_10"]})