Downloads · 30 days
30
1% of all-time downloads
ehdwns1516/klue-roberta-base-kornli
klue-roberta-base-kornli is a text classification model from ehdwns1516. Use it when you need a label for a piece of text. It is set up for transformers.
This model trained with Korean dataset. Input premise sentence and hypothesis sentence. You can use English, but don't expect accuracy. If the context is longer than 1200 characters, the context may be cut in the midd…
Downloads · 30 days
30
1% of all-time downloads
All-time downloads
2K
Public
Repo size
885 MB
Likes
1
Public
Click a slice to open those files.
.bin443 MB · 100%
From the Hugging Face model README
klue-roberta-base-kornli DEMO: Ainize DEMO
klue-roberta-base-kornli API: Ainize API
Language model: klue/roberta-base
Language: Korean
Training data: kakaobrain KorNLI
Eval data: kakaobrain KorNLI
Code: See Ainize Workspace
from transformers import AutoTokenizer, pipeline
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/klue-roberta-base-kornli")
classifier = pipeline(
"text-classification",
model="ehdwns1516/klue-roberta-base-kornli",
return_all_scores=True,
)
premise = "your premise"
hypothesis = "your hypothesis"
result = dict()
result[0] = classifier(premise + tokenizer.sep_token + hypothesis)[0]