Upload folder using huggingface_hub

app.py

@@ -791,7 +791,25 @@ def _build_grid_summary_figure():
     row_order = pivot.max(axis=1).sort_values(ascending=True).index.tolist()
     pivot = pivot.loc[row_order]
 
-    text = [[f"{v:.3f}" if pd.notna(v) else "" for v in row] for row in pivot.values]
+    # Embed star in the best cell's text (avoids categorical axis coordinate issues)
+    best_idx = agg["silhouette_cosine"].idxmax()
+    best = agg.loc[best_idx]
+    best_short = _short(best["model"])
+    best_thresh = best["threshold"]
+
+    rows_list = list(pivot.index)
+    cols_list = list(pivot.columns)
+    text = []
+    for ri, row_name in enumerate(rows_list):
+        row_text = []
+        for ci, col_val in enumerate(cols_list):
+            v = pivot.iloc[ri, ci]
+            if pd.notna(v):
+                cell = f"★ {v:.3f}" if (row_name == best_short and col_val == best_thresh) else f"{v:.3f}"
+            else:
+                cell = ""
+            row_text.append(cell)
+        text.append(row_text)
 
     fig = go.Figure(go.Heatmap(
         z=pivot.values,
@@ -805,17 +823,6 @@ def _build_grid_summary_figure():
         hovertemplate="Model: %{y}<br>Threshold: %{x}<br>Mean silhouette: %{z:.3f}<extra></extra>",
     ))
 
-    # Star annotation on the best cell
-    best = agg.loc[agg["silhouette_cosine"].idxmax()]
-    fig.add_annotation(
-        x=str(best["threshold"]),
-        y=_short(best["model"]),
-        text="★",
-        showarrow=False,
-        font=dict(size=18, color="#1a3a5c"),
-        xref="x", yref="y",
-    )
-
     n_models = len(pivot.index)
     fig.update_layout(
         title=dict(