colinglab

/

UMBERTO_fine-tuned_NERPA

Like 0

Follow

CoLingLab | Computational Linguistics Laboratory - University of Pisa 13

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("token-classification", model="colinglab/UMBERTO_fine-tuned_NERPA")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification

tokenizer = AutoTokenizer.from_pretrained("colinglab/UMBERTO_fine-tuned_NERPA")
model = AutoModelForTokenClassification.from_pretrained("colinglab/UMBERTO_fine-tuned_NERPA", device_map="auto")