fix: tokens dataframe header

app.py

@@ -3,6 +3,7 @@ from spacy import displacy
 import random
 from spacy.tokens import Span
 import gradio as gr
+import pandas as pd
 
 DEFAULT_MODEL = "en_core_web"
 DEFAULT_TEXT = "Apple is looking at buying U.K. startup for $1 billion."
@@ -57,6 +58,7 @@ def token(text, attributes, model):
         for attr in attributes:
             tok_data.append(getattr(tok, attr))
         data.append(tok_data)
+    data = pd.DataFrame(data, columns=attributes)
     return data
 
 
@@ -150,7 +152,8 @@ with demo:
             with gr.Column():
                 tok_input = gr.CheckboxGroup(
                     DEFAULT_TOK_ATTR, value=DEFAULT_TOK_ATTR)
-                tok_output = gr.Dataframe(overflow_row_behaviour="paginate")
+                tok_output = gr.Dataframe(
+                    headers=DEFAULT_TOK_ATTR, overflow_row_behaviour="paginate")
             tok_button = gr.Button("Generate this tab")
         with gr.TabItem("Similarity"):
             with gr.Row():

requirements.txt

@@ -1,4 +1,4 @@
-
+pandas==1.4.2
 gradio==3.0.18
 spacy==3.3.1