Update app.py

app.py

@@ -22,6 +22,8 @@ def compute_df():
             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
             metrics_cur = dic["metrics"]
             for k, v in metrics_cur.items():
                 dic.update({k: v["ndcg_at_10"]})
@@ -32,7 +34,7 @@ def compute_df():
 
     df = pd.DataFrame(metrics)
     df = df[cols]
-    df["model"] = df["model"].apply(lambda x: x.split("/")[-1])
+    # df["model"] = df["model"].apply(lambda x: x.split("/")[-1])
     # round all numeric columns
     # avg all numeric columns
     df["avg"] = df.iloc[:, 2:].mean(axis=1)