roberta_topic / README.md
pawlo2013's picture
Update README.md
7c06c8d verified
RoBERTa model finetuned for topic classification.
labels = ['books', 'movies', 'sports']