Update app.py

app.py

@@ -49,7 +49,7 @@ def get_hidden_states(raw_original_prompt):
     tokens = tokenizer.batch_decode(model_inputs.input_ids)
     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)
-    with gr.Row() as tokens_container:
+    # with gr.Row() as tokens_container:
         # for token in tokens:
         #     gr.Button(token)
     return str(tokens)