Update README.md

README.md

@@ -23,16 +23,15 @@ Multi-label binary sequence classification model developed by [Dejan Marketing](
 
 This model is an albert/albert-base-v2 fine-tune trained to return an array of values for 10 classification labels.
 
-# Human-readable labels
-label_map = {
-    1: 'Commercial',
-    2: 'Non-Commercial',
-    3: 'Branded',
-    4: 'Non-Branded',
-    5: 'Informational',
-    6: 'Navigational',
-    7: 'Transactional',
-    8: 'Commercial Investigation',
-    9: 'Local',
-    10: 'Entertainment'
-}
+# Labels
+
+    LABEL_0: 'Commercial',
+    LABEL_1: 'Non-Commercial',
+    LABEL_2: 'Branded',
+    LABEL_3: 'Non-Branded',
+    LABEL_4: 'Informational',
+    LABEL_5: 'Navigational',
+    LABEL_6: 'Transactional',
+    LABEL_7: 'Commercial Investigation',
+    LABEL_8: 'Local',
+    LABEL_9: 'Entertainment'