afg1 commited on
Commit
aae4abd
1 Parent(s): f8084f8

Update config with id2label

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -40,5 +40,9 @@
40
  "torch_dtype": "float32",
41
  "transformers_version": "4.36.2",
42
  "type_vocab_size": 1,
43
- "vocab_size": 50265
 
 
 
 
44
  }
 
40
  "torch_dtype": "float32",
41
  "transformers_version": "4.36.2",
42
  "type_vocab_size": 1,
43
+ "vocab_size": 50265,
44
+ "id2label": {
45
+ "0": "NOT_DISEASE",
46
+ "1": "DISEASE"
47
+ }
48
  }