CafeBERT Fine-tuned for Toxic Classification
This model is a fine-tuned version of CafeBERT on a toxic comment classification task.
Usage
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
- Downloads last month
- 2
Model tree for UngLong/cafebert-classification-ft
Base model
uitnlp/CafeBERT