ksgr5566/ner
Preview • Updated • 17
How to use enigma-112/akai_ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="enigma-112/akai_ner") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("enigma-112/akai_ner")
model = AutoModelForTokenClassification.from_pretrained("enigma-112/akai_ner")The model recognizes 3 types of entities - pest names, seed types, crop names.
Helps extracting the crop name, pest names and seed details from a query asked.