# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("pawlo2013/roberta_topic")
model = AutoModelForSequenceClassification.from_pretrained("pawlo2013/roberta_topic")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
RoBERTa model finetuned for topic classification.
labels = ['books', 'movies', 'sports']
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="pawlo2013/roberta_topic")