Views
No views yet
1from transformers import BertForSequenceClassification, BertTokenizer
2
3# 모델과 토크나이저 로드
4model_name = "rmsdud/kobert-classifier"
5tokenizer = BertTokenizer.from_pretrained(model_name)
6model = BertForSequenceClassification.from_pretrained(model_name)
7
8# 추론
9text = "분류할 텍스트를 입력하세요."
10inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
11outputs = model(**inputs)
12logits = outputs.logits
13predicted_class = logits.argmax(-1).item()
14print(f"예측 클래스: predicted_class")