add headers

app.py

@@ -28,7 +28,7 @@ def extract_pii(text):
     pii_entities = res.json()["results"]["documents"][0]["entities"]
     return list(map(lambda row: [row["text"], row["category"], row["confidenceScore"]], pii_entities))
 
-demo = gr.Interface(extract_pii, inputs="textbox", outputs="dataframe")
+demo = gr.Interface(extract_pii, inputs="textbox", outputs=gr.DataFrame(headers=["Text", "Category", "Confidence Score"]))
 
 if __name__ == "__main__":
     demo.launch()