ludocomito commited on
Commit
5d7f616
·
1 Parent(s): ba0d0ab

correct attention patterns

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- is_row_active = step == "intra" and row_idx == 2
199
- is_col_active = step == "inter" and col_idx == 2
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]