hf_model.py

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']