roberta_topic / README.md
pawlo2013's picture
Update README.md
7c06c8d verified

RoBERTa model finetuned for topic classification.

labels = ['books', 'movies', 'sports']