How to use moo3030/roberta-for-pii with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="moo3030/roberta-for-pii")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("moo3030/roberta-for-pii") model = AutoModelForTokenClassification.from_pretrained("moo3030/roberta-for-pii", device_map="auto")
PII Redactor model using RoBERTa. The goal of this model is to remove personally identifiable information from text documents.
PII includes:
Base model
