from transformers import AutoTokenizer, AutoModelForSequenceClassification,pipeline import torch def load_model(): model_name = "MoritzLaurer/DeBERTa-v3-base-mnli" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForSequenceClassification.from_pretrained(model_name) classifier = pipeline(task='zero-shot-classification', model=model, tokenizer=tokenizer, framework='pt') return classifier def classifier_zero(classifier,sequence:str,labels:list,multi_class:bool): outputs=classifier(sequence, labels,multi_label=multi_class) return outputs['labels'], outputs['scores']