Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ from transformers import BertTokenizer, MegatronBertModel
|
|
47 |
tokenizer = BertTokenizer.from_pretrained('NYTK/PULI-BERT-Large')
|
48 |
model = MegatronBertModel.from_pretrained('NYTK/PULI-BERT-Large')
|
49 |
text = "Replace me by any text you'd like."
|
50 |
-
encoded_input = tokenizer(text, return_tensors='pt')
|
51 |
output = model(**encoded_input)
|
52 |
|
53 |
```
|
|
|
47 |
tokenizer = BertTokenizer.from_pretrained('NYTK/PULI-BERT-Large')
|
48 |
model = MegatronBertModel.from_pretrained('NYTK/PULI-BERT-Large')
|
49 |
text = "Replace me by any text you'd like."
|
50 |
+
encoded_input = tokenizer(text, return_tensors='pt', do_lower_case=False)
|
51 |
output = model(**encoded_input)
|
52 |
|
53 |
```
|