Small change

app.py

@@ -3,8 +3,17 @@ from transformers import pipeline
 
 app = Flask(__name__)
 
-# Load the text classification pipeline
-classifier = pipeline("text-classification", model="Davephoenix/bert-bullying-detector")
+# Define label mappings
+id2label = {0: "not_bullying", 1: "bullying"}
+label2id = {"not_bullying": 0, "bullying": 1}
+
+# Load the text classification pipeline with label mappings
+classifier = pipeline(
+    "text-classification",
+    model="Davephoenix/bert-bullying-detector",
+    id2label=id2label,
+    label2id=label2id,
+)
 
 
 @app.route("/classify/text", methods=["POST"])