Views
No views yet
| tag | token |
|---|---|
| B-ITEM | BEGINNING ITEM |
| I-ITEM | INSIDE ITEM |
| B-METRIC | BEGINNING METRIC |
| I-METRIC | INSIDE METRIC |
| O | OUTSIDE |
1from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
2tokenizer = AutoTokenizer.from_pretrained("Wanjiru/autotrain_gro_ner")
3
4model = AutoModelForTokenClassification.from_pretrained("Wanjiru/autotrain_gro_ner")
5
6nlp = pipeline("ner", model=model, tokenizer=tokenizer)
7example = "Wanjru"
8ner_res = nlp(example)
9