Update app.py

app.py

@@ -98,7 +98,7 @@ def get_hidden_states(raw_original_prompt, force_hidden_states=False):
         # TODO: document this!
         important_tokens = (1 + 
                             F.normalize(hidden_states, dim=-1).diff(dim=0)
-                            .norm(dim=-1).topk(k=7).indices[:, 0].cpu().numpy())
+                            .norm(dim=-1).topk(k=5).indices[:, 0].cpu().numpy())
         print(f'{important_tokens=}')
         global_state.local_state.hidden_states = hidden_states.cpu().detach()