Update app.py

app.py

@@ -10,13 +10,13 @@ classifier = pipeline(
 )
 
 # Label mapping (MUST match training)
-id_to_label = {
-    0: "administrative_action",
-    1: "attendance",
-    2: "check_in",
-    3: "clarification",
-    4: "general_chat",
-    5: "technical_help"
+label_map = {
+    "LABEL_0": "administrative_action",
+    "LABEL_1": "attendance",
+    "LABEL_2": "check_in",
+    "LABEL_3": "clarification",
+    "LABEL_4": "general_chat",
+    "LABEL_5": "technical_help"
 }
 
 def classify_request(role, context, request):
@@ -26,9 +26,11 @@ def classify_request(role, context, request):
     
     # Get highest scoring label
     best = max(outputs, key=lambda x: x["score"])
+
+    readable_label = label_map.get(best["label"], best["label"])
     
     return {
-        "Predicted intent": best["label"],
+        "Predicted intent": readable_label,
         "Confidence": round(best["score"], 3)
     }