new tag 103a08f
scasps commited on
How to use Wyona/message-classification-question-other-smalltalk-modified with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Wyona/message-classification-question-other-smalltalk-modified") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Wyona/message-classification-question-other-smalltalk-modified")
model = AutoModelForSequenceClassification.from_pretrained("Wyona/message-classification-question-other-smalltalk-modified", device_map="auto")