Update config.json

config.json

@@ -1,41 +1,53 @@
 {
-  "_name_or_path": "tiiuae/falcon-7b",
-  "alibi": false,
-  "apply_residual_connection_post_layernorm": false,
-  "architectures": [
-    "FalconForCausalLM"
-  ],
-  "attention_dropout": 0.0,
-  "auto_map": {
-    "AutoConfig": "tiiuae/falcon-7b--configuration_falcon.FalconConfig",
-    "AutoModel": "tiiuae/falcon-7b--modeling_falcon.FalconModel",
-    "AutoModelForCausalLM": "tiiuae/falcon-7b--modeling_falcon.FalconForCausalLM",
-    "AutoModelForQuestionAnswering": "tiiuae/falcon-7b--modeling_falcon.FalconForQuestionAnswering",
-    "AutoModelForSequenceClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForSequenceClassification",
-    "AutoModelForTokenClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForTokenClassification"
-  },
-  "bias": false,
-  "bos_token_id": 11,
-  "custom_pipelines": {
-    "text-generation": {
-      "pt": "AutoModelForCausalLM"
-    }
-  },
-  "eos_token_id": 11,
-  "hidden_dropout": 0.0,
-  "hidden_size": 4544,
-  "initializer_range": 0.02,
-  "layer_norm_epsilon": 1e-05,
-  "model_type": "falcon",
-  "multi_query": true,
-  "new_decoder_architecture": false,
-  "num_attention_heads": 71,
-  "num_hidden_layers": 32,
-  "num_kv_heads": 71,
-  "pad_token_id": 11,
-  "parallel_attn": true,
-  "torch_dtype": "bfloat16",
-  "transformers_version": "4.37.1",
-  "use_cache": true,
-  "vocab_size": 65024
-}
+    "_name_or_path": "tiiuae/falcon-7b",
+    "alibi": false,
+    "apply_residual_connection_post_layernorm": false,
+    "architectures": [
+      "FalconForCausalLM"
+    ],
+    "attention_dropout": 0.0,
+    "auto_map": {
+      "AutoConfig": "tiiuae/falcon-7b--configuration_falcon.FalconConfig",
+      "AutoModel": "tiiuae/falcon-7b--modeling_falcon.FalconModel",
+      "AutoModelForCausalLM": "tiiuae/falcon-7b--modeling_falcon.FalconForCausalLM",
+      "AutoModelForQuestionAnswering": "tiiuae/falcon-7b--modeling_falcon.FalconForQuestionAnswering",
+      "AutoModelForSequenceClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForSequenceClassification",
+      "AutoModelForTokenClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForTokenClassification"
+    },
+    "bias": false,
+    "bos_token_id": 11,
+    "custom_pipelines": {
+      "text-generation": {
+        "pt": "AutoModelForCausalLM"
+      }
+    },
+    "eos_token_id": 11,
+    "hidden_dropout": 0.0,
+    "hidden_size": 4544,
+    "initializer_range": 0.02,
+    "layer_norm_epsilon": 1e-05,
+    "model_type": "falcon",
+    "multi_query": true,
+    "new_decoder_architecture": false,
+    "num_attention_heads": 71,
+    "num_hidden_layers": 32,
+    "num_kv_heads": 71,
+    "pad_token_id": 11,
+    "parallel_attn": true,
+    "torch_dtype": "bfloat16",
+    "transformers_version": "4.37.1",
+    "use_cache": true,
+    "vocab_size": 65024,
+    "id2label": {
+        "0": "EXPLOSIVES AND BREAKING AGENTS",
+        "1": "EXPLODING VESSELS UNDER PRESSURE",
+        "2": "TOXIC",
+        "3": "ENTRAPMENT",
+        "4": "INUNDATION",
+        "5": "SLIP OR FALL OF PERSON",
+        "6": "HANDLING OF MATERIALS",
+        "7": "HAULAGE",
+        "8": "ELECTRICAL",
+        "9": "FIRE/EXPLOSION"
+      }
+  }