conviette commited on
Commit
7573721
·
1 Parent(s): 8b6c0f4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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('model\\')
25
- model = BertForSequenceClassification.from_pretrained('model\\')
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')