File size: 231 Bytes
c7bffdb
 
 
 
 
1
2
3
4
5
from transformers import pipeline
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")

categories = ['lifestyle', 'technology', 'entertainment', 'news']
classifier(text, categories, multi_class=True)