File size: 227 Bytes
07b50c0
 
 
 
 
1
2
3
4
5
from transformers import pipeline

# Load once and reuse
zero_shot = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
ner = pipeline("ner", model="dslim/bert-base-NER", aggregation_strategy="simple")