Spaces:
Runtime error
Runtime error
File size: 634 Bytes
17a8518 6ec105b 3b5b50d 17a8518 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification,pipeline
import torch
def load_model():
model_name = "MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli"
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']
|