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
- 8