# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("Afreen/ner_test")
model = AutoModelForTokenClassification.from_pretrained("Afreen/ner_test")Quick Links
About the Model
An Environmental Named Entity Recognition model, trained on dataset from USEPA to recognize environmental due diligence (7 entities) from a given text corpus (remediation reports, record of decision, 5 year record etc). This model was built on top of distilbert-base-uncased
- Dataset: https://data.mendeley.com/datasets/tx6vmd4g9p/4
- Dataset Reasearch Paper: https://doi.org/10.1016/j.dib.2022.108579
- Downloads last month
- 6
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="Afreen/ner_test")