Spaces:
Sleeping
Sleeping
Commit ·
5d7f616
1
Parent(s): ba0d0ab
correct attention patterns
Browse files
app.py
CHANGED
|
@@ -195,8 +195,8 @@ def generate_grid_html(step, masked_indices=None):
|
|
| 195 |
row_idx = i // 5
|
| 196 |
col_idx = i % 5
|
| 197 |
is_masked = i in masked_indices
|
| 198 |
-
|
| 199 |
-
|
| 200 |
|
| 201 |
# Determine cell color based on column
|
| 202 |
base_color = colors[col_idx]
|
|
|
|
| 195 |
row_idx = i // 5
|
| 196 |
col_idx = i % 5
|
| 197 |
is_masked = i in masked_indices
|
| 198 |
+
is_col_active = step == "intra" and col_idx == 2 # Intra-cellular highlights columns
|
| 199 |
+
is_row_active = step == "inter" and row_idx == 2 # Inter-cellular highlights rows
|
| 200 |
|
| 201 |
# Determine cell color based on column
|
| 202 |
base_color = colors[col_idx]
|