gg4ever/intent-classification-korean

from transformers import TextClassificationPipeline
from transformers import AutoModelForSequenceClassification, AutoTokenizer

model_path = "gg4ever/intent-classifcation-korean"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)

text_classifier = TextClassificationPipeline(
    tokenizer=tokenizer, 
    model=model.to('cpu'), 
    return_all_scores=True
)

# predict
text = "이름이 뭐에요?"

preds_list = text_classifier(text)
preds_list