Add few-shot info in graphs

app.py

@@ -321,9 +321,8 @@ def plot_size_vs_score(df: pd.DataFrame, hidden_df: pd.DataFrame) -> go.Figure:
     df2 = df2[df2["#Params (B)"] > 0]
     df2 = df2[["model_name_for_query", "#Params (B)", "AVG", "Few-shot"]]
     df2["AVG"] = df2["AVG"].astype(float)
-    df2["Few-shot"] = df2["Few-shot"].astype(int)
     df2 = df2.rename(columns={"model_name_for_query": "Model", "Few-shot": "n-shot"})
-    df2["model_name_without_org_name"] = df2["Model"].str.split("/").str[-1]
+    df2["model_name_without_org_name"] = df2["Model"].str.split("/").str[-1] + " (" + df2["n-shot"] + "-shot)"
     fig = px.scatter(
         df2,
         x="#Params (B)",