Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ from transformers import (AutoModelForTokenClassification,
|
|
44 |
|
45 |
model_checkpoint = "jsylee/scibert_scivocab_uncased-finetuned-ner"
|
46 |
model = AutoModelForTokenClassification.from_pretrained(model_checkpoint, num_labels=5,
|
47 |
-
id2label={0: 'O', 1: 'B-DRUG', 2: 'I-DRUG', 3: 'B-
|
48 |
)
|
49 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
50 |
|
|
|
44 |
|
45 |
model_checkpoint = "jsylee/scibert_scivocab_uncased-finetuned-ner"
|
46 |
model = AutoModelForTokenClassification.from_pretrained(model_checkpoint, num_labels=5,
|
47 |
+
id2label={0: 'O', 1: 'B-DRUG', 2: 'I-DRUG', 3: 'B-EFFECT', 4: 'I-EFFECT'}
|
48 |
)
|
49 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
50 |
|