Spaces:
Runtime error
Runtime error
from transformers import pipeline | |
umbrellaSubjects = [ | |
"Science, Technology, Engineering, Mathematics", | |
"Philosophy", | |
"Arts and Humanities", | |
"Social Sciences", | |
"Languages", | |
"Professional Studies" | |
] | |
classifier = pipeline("zero-shot-classification", model="vicgalle/xlm-roberta-large-xnli-anli") | |
def process_string(string): | |
# Remove whitespace from string | |
string = string.replace(' ', '') | |
# Split string by comma | |
string_list = string.split(',') | |
# Return list with no whitespace | |
return [term.strip() for term in string_list] | |
def classify(sentences,categories): | |
return classifier(process_string(sentences),process_string(categories)) | |