Update config.json

config.json

@@ -8,30 +8,30 @@
  "hidden_dropout_prob": 0.1,
  "hidden_size": 768,
  "id2label": {
- "0": "LABEL_0",
- "1": "LABEL_1",
- "2": "LABEL_2",
- "3": "LABEL_3",
- "4": "LABEL_4",
- "5": "LABEL_5",
- "6": "LABEL_6",
- "7": "LABEL_7",
- "8": "LABEL_8",
- "9": "LABEL_9"
+ "0": "sexual",
+ "1": "hate",
+ "2": "violence",
+ "3": "harassment",
+ "4": "self-harm",
+ "5": "sexual/minors",
+ "6": "hate/threatening",
+ "7": "violence/graphic",
+ "8": "fake news",
+ "9": "OK"
  },
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "label2id": {
- "LABEL_0": 0,
- "LABEL_1": 1,
- "LABEL_2": 2,
- "LABEL_3": 3,
- "LABEL_4": 4,
- "LABEL_5": 5,
- "LABEL_6": 6,
- "LABEL_7": 7,
- "LABEL_8": 8,
- "LABEL_9": 9
+ "sexual": 0,
+ "hate": 1,
+ "violence": 2,
+ "harassment": 3,
+ "self-harm": 4,
+ "sexual/minors": 5,
+ "hate/threatening": 6,
+ "violence/graphic": 7,
+ "fake news": 8,
+ "OK": 9
  },
  "layer_norm_eps": 1e-07,
  "max_position_embeddings": 512,