Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model_name = "ssunbear/bert-base-finetuned-ynat"
4
5# 모델과 토크나이저 로드
6tokenizer = AutoTokenizer.from_pretrained("klue/bert-base")
7model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=7)