Error while running Named Entity Recognition
- Dominant language
- Python
- Stars
- 1.7k
- Forks
- 232
- PR merge metrics
- No merged PRs in 30d
Description
I tried using SciBERT for NER using the following command-
from transformers import *
tokenizer = AutoTokenizer.from_pretrained('allenai/scibert_scivocab_uncased')
model = AutoModel.from_pretrained('allenai/scibert_scivocab_uncased')
nlp = pipeline('ner',model = model,tokenizer=tokenizer)
nlp('Clinical features of culture-proven Mycoplasma pneumoniae infections at King Abdulaziz University Hospital, Jeddah, Saudi Arabia')
While running it on a sample sentence, I get the following error-
Traceback (most recent call last):
File "", line 1, in
File "/python3.6/site-packages/transformers/pipelines.py", line 927, in __call__
for idx, label_idx in enumerate(labels_idx)
File "/python3.6/site-packages/transformers/pipelines.py", line 928, in
if self.model.config.id2label[label_idx] not in self.ignore_labels
KeyError: 422
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.