# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("JunHwi/kold_binary")
model = AutoModelForSequenceClassification.from_pretrained("JunHwi/kold_binary")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Pretraining KoLD Dataset with pretrained "koelectra-v3" model.
dataset : https://github.com/boychaboy/KOLD
pretrained_model : https://huggingface.co/monologg/koelectra-base-v3-discriminator
So you should use tokenizer with "koelectra-base-v3-discriminator".
label maps are like
{0: "not_hate_speech", 1: "hate_speech"}
- Downloads last month
- 10
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="JunHwi/kold_binary")