Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import string
 import re
+import pandas as pd
 
 from transformers import pipeline
 
@@ -33,7 +34,20 @@ def parse_postal_address_can(text):
     text = text.lower()
     text = replace_punctuation_with_space(text)
     text = replace_multiple_spaces(text)
-    return token_classifier(text)
+    results = token_classifier(text)
+
+    # Format the output as a dataframe
+    data = []
+    for result in results:
+        data.append({
+            'entity_group': result['entity_group'],
+            'score': round(result['score'], ndigits=2),
+            'word': result['word'],
+            'start': result['start'],
+            'end': result['end']
+        })
+    df = pd.DataFrame(data)
+    return df
 
 
 #
@@ -51,14 +65,12 @@ with gr.Blocks() as demo:
         label="Canadian postal address",
         render=False
     )
-    output_text = gr.Textbox(lines=10, render=False)
+    output = gr.DataFrame(row_count=10, render=False)
 
     gr.Interface(
         fn=parse_postal_address_can,
         inputs=[input_text,],
-        outputs=[output_text,],
-        allow_flagging="never"
-        #clear_btn=None
+        outputs=[output,]
     )
 
     with gr.Accordion("Documentation", open=False):
@@ -74,5 +86,5 @@ with gr.Blocks() as demo:
                   no training data with "street", "str.", ...
         """)
 
-if __name__ == "__main__":
-    demo.launch()
+
+demo.launch()