Update

app.py

@@ -312,7 +312,7 @@ def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
         df,
         id_vars=["Model", "model_name_without_org_name", "#Params (B)", "n-shot"],
         value_vars=AVG_COLUMNS,
-        var_name="Task_Category",
+        var_name="Category",
         value_name="Score",
     )
     fig = px.scatter(
@@ -320,8 +320,8 @@ def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
         x="#Params (B)",
         y="Score",
         text="model_name_without_org_name",
-        color="Task_Category",
-        hover_data=["Model", "n-shot", "Task_Category"],
+        color="Category",
+        hover_data=["Model", "n-shot", "Category"],
     )
     fig.update_traces(
         hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>n-shot: %{customdata[1]}<br>%{customdata[2]}: %{y:.4f}<extra></extra>",