File size: 155 Bytes
0dd2eb3
 
 
 
 
 
1
2
3
4
5
6
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="priyabrat/bert_categorisation")

)