Spaces:
Running
Running
Marlin Lee commited on
Commit ·
68497cf
1
Parent(s): 5891413
Sync explorer_app.py and clip_utils.py from main repo
Browse files- scripts/explorer_app.py +11 -3
scripts/explorer_app.py
CHANGED
|
@@ -666,6 +666,14 @@ def make_cross_sae_comparison_html(ds_a, feat_a, ds_b, feat_b, n=4, size=160):
|
|
| 666 |
items_a = _collect(ds_a, feat_a)
|
| 667 |
items_b = _collect(ds_b, feat_b)
|
| 668 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 669 |
def _grid_html(items, label, color):
|
| 670 |
header = (f'<div style="background:{color};color:#fff;font-size:11px;font-weight:bold;'
|
| 671 |
f'text-align:center;padding:4px;border-radius:4px;margin-bottom:6px">{label}</div>')
|
|
@@ -677,10 +685,10 @@ def make_cross_sae_comparison_html(ds_a, feat_a, ds_b, feat_b, n=4, size=160):
|
|
| 677 |
f' style="border:1px solid #ccc;border-radius:3px;display:block"/>'
|
| 678 |
f'<div style="font-size:9px;color:#555;margin-top:2px">{cap}</div></div>')
|
| 679 |
grid += '</div>'
|
| 680 |
-
return
|
| 681 |
|
| 682 |
-
label_a = f"{ds_a['label']} — feat {feat_a}"
|
| 683 |
-
label_b = f"{ds_b['label']} — feat {feat_b}"
|
| 684 |
col_a = _grid_html(items_a, label_a, "#2563a8")
|
| 685 |
col_b = _grid_html(items_b, label_b, "#b85c00")
|
| 686 |
|
|
|
|
| 666 |
items_a = _collect(ds_a, feat_a)
|
| 667 |
items_b = _collect(ds_b, feat_b)
|
| 668 |
|
| 669 |
+
def _strip_dim(label):
|
| 670 |
+
"""Remove parenthetical dimension info like '(d=32K)' or '(d=32K, k=160)'."""
|
| 671 |
+
out = label
|
| 672 |
+
while '(' in out and ')' in out:
|
| 673 |
+
l, r = out.index('('), out.index(')')
|
| 674 |
+
out = out[:l].rstrip() + out[r+1:]
|
| 675 |
+
return out.strip(' —').strip()
|
| 676 |
+
|
| 677 |
def _grid_html(items, label, color):
|
| 678 |
header = (f'<div style="background:{color};color:#fff;font-size:11px;font-weight:bold;'
|
| 679 |
f'text-align:center;padding:4px;border-radius:4px;margin-bottom:6px">{label}</div>')
|
|
|
|
| 685 |
f' style="border:1px solid #ccc;border-radius:3px;display:block"/>'
|
| 686 |
f'<div style="font-size:9px;color:#555;margin-top:2px">{cap}</div></div>')
|
| 687 |
grid += '</div>'
|
| 688 |
+
return f'<div style="display:flex;flex-direction:column">{header}{grid}</div>'
|
| 689 |
|
| 690 |
+
label_a = f"{_strip_dim(ds_a['label'])} — feat {feat_a}"
|
| 691 |
+
label_b = f"{_strip_dim(ds_b['label'])} — feat {feat_b}"
|
| 692 |
col_a = _grid_html(items_a, label_a, "#2563a8")
|
| 693 |
col_b = _grid_html(items_b, label_b, "#b85c00")
|
| 694 |
|