Spaces:
Sleeping
Sleeping
File size: 358 Bytes
61d8616 be5e682 61d8616 be5e682 |
1 2 3 4 5 6 7 8 9 |
# 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)
|