Hugging Face's logo

ssunbear
/
bert-base-finetuned-ynat-v2

Text Classification
Safetensors
Korean
bert
Model card Files Community
ssunbear commited on
Commit
6ea78f2
•
1 Parent(s): 1ca4c06

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,8 +33,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
33
  model_name = "ssunbear/bert-base-finetuned-ynat-v2"
34
 
35
  # 모델과 토크나이저 로드
36
- tokenizer = AutoTokenizer.from_pretrained(model_name)
37
- model = AutoModelForSequenceClassification.from_pretrained(model_name)
38
  ```
39
 
40
  ## 훈련 데이터
 
33
  model_name = "ssunbear/bert-base-finetuned-ynat-v2"
34
 
35
  # 모델과 토크나이저 로드
36
+ tokenizer = AutoTokenizer.from_pretrained("klue/bert-base")
37
+ model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=7)
38
  ```
39
 
40
  ## 훈련 데이터