First version of the distilbert_classifier2 model and tokenizer. d198d63
Simone Gayed Said commited on
How to use nlpunibo/distilbert_classifier2 with Transformers:
# Load model directly
from transformers import AutoTokenizer, DistilBertClassifier
tokenizer = AutoTokenizer.from_pretrained("nlpunibo/distilbert_classifier2")
model = DistilBertClassifier.from_pretrained("nlpunibo/distilbert_classifier2", device_map="auto")