added labeling

app/model_loader.py

@@ -17,8 +17,16 @@ model.eval()
 
 print("Model loaded successfully.")
 
-# Load label mapping from model repo
-label_mapping = model.config.id2label if hasattr(model.config, "id2label") else None
+
+label_mapping = {
+    0: "Delayed payment (no dispute)",
+    1: "Quality dispute",
+    2: "No formal contract",
+    3: "Partial payment dispute",
+    4: "Government procurement delay",
+    5: "Service-related dispute"
+}
+
 
 if label_mapping is None or len(label_mapping) == 0:
     # fallback if id2label not saved in config