metadata
language:
- en
Open Named Entity Recognition (English)
You can read the details in this post.
from transformers import pipeline
nlp = pipeline('token-classification', 'yongsun-yoon/deberta-v3-base-open-ner', aggregation_strategy='simple')
text = 'Heat the olive oil in a frying pan, add the onion and cook for 5 minutes until softened and starting to turn golden. Set aside.'
entity_type = 'ingredient'
input_text = f'{text}{nlp.tokenizer.sep_token}{entity_type}' # [SEP]
nlp(input_text) # olive oil, onion