Update app.py

app.py

@@ -33,11 +33,14 @@ def predict_toxicity(comment):
     
     return {labels[i]: float(probabilities[i]) for i in range(len(labels))}
 
+import pandas as pd
+
 def format_toxicity_data(comment):
     """Formats the toxicity scores for a modern bar graph."""
     scores = predict_toxicity(comment)
-    data = [[label, score] for label, score in scores.items()]
-    return data
+    df = pd.DataFrame(list(scores.items()), columns=["Category", "Score"])
+    return df
+
 
 # Gradio interface
 demo = gr.Interface(