Update README.md

README.md

@@ -31,18 +31,19 @@ A custom synthetic dataset was created for this task, focusing on the structural
 
 ### Categories:
 
-1- Shipping and Delivery
-
-2- Customer Service
-
-3- Price and Value
-
-4- Quality and Performance
-
-5- Use and Design
-
-6- Other
-
+### label_mapping = {
+    "shipping_and_delivery": 0,
+    
+    "customer_service": 1,
+    
+    "price_and_value": 2,
+    
+    "quality_and_performance": 3,
+    
+    "use_and_design": 4,
+    
+    "other": 5
+}
 ### Training Hyperparameters
 Model: distilbert/distilbert-base-uncased