Update app.py

app.py

@@ -97,6 +97,7 @@ def get_hidden_states(raw_original_prompt, force_hidden_states=False):
     tokens = tokenizer.batch_decode(model_inputs.input_ids[0])
     if global_state.wait_with_hidden_states and not force_hidden_states:
         global_state.local_state.hidden_states = None
+        important_tokens = [] # cannot find important tokens without the hidden states
     else:
         outputs = model(**model_inputs, output_hidden_states=True, return_dict=True)
         hidden_states = torch.stack([h.squeeze(0).cpu().detach() for h in outputs.hidden_states], dim=0)