abratnap commited on
Commit
0b27ad8
1 Parent(s): 8f9b0f8

Update config with labels information

Browse files

Add id2label and label2id labels to represent classes.
This is one of the requirement to run model with huggingface text embedding inference.

Files changed (1) hide show
  1. config.json +9 -1
config.json CHANGED
@@ -23,5 +23,13 @@
23
  "transformers_version": "4.31.0",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
- "vocab_size": 50265
 
 
 
 
 
 
 
 
27
  }
 
23
  "transformers_version": "4.31.0",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
+ "vocab_size": 50265,
27
+ "id2label": {
28
+ "0": "no_HAP",
29
+ "1": "HAP"
30
+ },
31
+ "label2id": {
32
+ "no_HAP": 0,
33
+ "HAP": 1
34
+ }
35
  }