Update README.md
Browse files
README.md
CHANGED
|
@@ -21,8 +21,8 @@ from KorBertTokenizer import KorBertTokenizer
|
|
| 21 |
from transformers import BertForSequenceClassification
|
| 22 |
import torch
|
| 23 |
|
| 24 |
-
tokenizer = KorBertTokenizer.from_pretrained('
|
| 25 |
-
model = BertForSequenceClassification.from_pretrained('
|
| 26 |
|
| 27 |
def classify(text):
|
| 28 |
inputs = tokenizer(text, padding='max_length', max_length=70, return_tensors='pt')
|
|
|
|
| 21 |
from transformers import BertForSequenceClassification
|
| 22 |
import torch
|
| 23 |
|
| 24 |
+
tokenizer = KorBertTokenizer.from_pretrained('conviette/korPolBERT')
|
| 25 |
+
model = BertForSequenceClassification.from_pretrained('conviette/korPolBERT')
|
| 26 |
|
| 27 |
def classify(text):
|
| 28 |
inputs = tokenizer(text, padding='max_length', max_length=70, return_tensors='pt')
|