# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("GautamR/akai_ner")
model = AutoModelForTokenClassification.from_pretrained("GautamR/akai_ner", device_map="auto")Quick Links
Model Card for Model ID
The model recognizes 3 types of entities - pest names, seed types, crop names.
Model Details
Model Description
- Developed by: SamagraX | Transforming Governance
- Shared by [optional]: SamagraX | Transforming Governance
- Model type: Named Entity Recognition for agriculture
- Language(s) (NLP): Python
- License: MIT
- Finetuned from model [optional]: distilbert-base-uncased
Uses
Helps extracting the crop name, pest names and seed details from a query asked.
- Downloads last month
- 8
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="GautamR/akai_ner")