RoBERTa Plant NER Model
Fine-tuned RoBERTa model for plant entity recognition.
Labels
O: Outside entityB-PLANT_COMMON: Beginning of common plant nameI-PLANT_COMMON: Inside common plant nameB-PLANT_SCI: Beginning of scientific plant nameI-PLANT_SCI: Inside scientific plant name
Usage
Python
from transformers import pipeline
classifier = pipeline("token-classification", "Dudeman523/roberta-plant-ner-fixed")
result = classifier("I have a Rosa damascena plant")
JavaScript (transformers.js)
import { pipeline } from '@xenova/transformers';
const classifier = await pipeline('token-classification', 'Dudeman523/roberta-plant-ner-fixed');
const result = await classifier('I have a Rosa damascena plant');
- Downloads last month
- 4