Update app.py

app.py

@@ -53,7 +53,7 @@ def get_hidden_states(raw_original_prompt):
     hidden_states = torch.stack([h.squeeze(0).cpu().detach() for h in outputs.hidden_states], dim=0)
     token_btns = ([gr.Button(token, visible=True) for token in tokens] 
                   + [gr.Button('', visible=False) for _ in range(MAX_PROMPT_TOKENS - len(tokens))])
-    return [hidden_state, *token_btns]
+    return [hidden_states, *token_btns]
 
 
 def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens, do_sample,