Add model name label to the graph

app.py

@@ -322,8 +322,18 @@ def plot_size_vs_score(df: pd.DataFrame, hidden_df: pd.DataFrame) -> go.Figure:
     df2 = df2[["model_name_for_query", "#Params (B)", "AVG"]]
     df2["AVG"] = df2["AVG"].astype(float)
     df2 = df2.rename(columns={"model_name_for_query": "Model"})
-    fig = px.scatter(df2, x="#Params (B)", y="AVG", hover_data=["Model"])
-    fig.update_traces(hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>AVG: %{y:.4f}<extra></extra>")
+    df2["model_name_without_org_name"] = df2["Model"].str.split("/").str[-1]
+    fig = px.scatter(
+        df2,
+        x="#Params (B)",
+        y="AVG",
+        text="model_name_without_org_name",
+        hover_data=["Model"],
+    )
+    fig.update_traces(
+        hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>AVG: %{y:.4f}<extra></extra>",
+        textposition="top right",
+    )
     fig.update_layout(yaxis_range=[0, 1])
     return fig