Small change

app.py

@@ -3,16 +3,15 @@ from transformers import pipeline
 
 app = Flask(__name__)
 
-# Define label mappings
-id2label = {0: "not_bullying", 1: "bullying"}
-label2id = {"not_bullying": 0, "bullying": 1}
+labels = {
+    "LABEL_0": "not_bullying",
+    "LABEL_1": "bullying",
+}
 
 # Load the text classification pipeline with label mappings
 classifier = pipeline(
     "text-classification",
     model="Davephoenix/bert-bullying-detector",
-    id2label=id2label,
-    label2id=label2id,
 )
 
 
@@ -23,6 +22,9 @@ def classify_text():
         return jsonify({"error": "Missing 'text' in request body"}), 400
     try:
         result = classifier(data["text"])
+        # Map the labels to their meanings
+        for item in result:
+            item["label"] = labels.get(item["label"], item["label"])
         return jsonify(result)
     except Exception as e:
         return jsonify({"error": str(e)}), 500