formal_project / app.py
stats-dev
test
61d8616
raw
history blame
358 Bytes
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("j5ng/kcbert-formal-classifier")
model = AutoModelForSequenceClassification.from_pretrained("j5ng/kcbert-formal-classifier")
formal_classifier = pipeline(task="text-classification", model=model, tokenizer=tokenizer)