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