Update README.md
Browse files
README.md
CHANGED
@@ -44,8 +44,8 @@ If you use this model, please cite the following paper:
|
|
44 |
```python
|
45 |
from transformers import BertTokenizer, MegatronBertModel
|
46 |
|
47 |
-
tokenizer = BertTokenizer.from_pretrained('NYTK/PULI-BERT-
|
48 |
-
model = MegatronBertModel.from_pretrained('NYTK/PULI-BERT-
|
49 |
text = "Replace me by any text you'd like."
|
50 |
encoded_input = tokenizer(text, return_tensors='pt')
|
51 |
output = model(**encoded_input)
|
|
|
44 |
```python
|
45 |
from transformers import BertTokenizer, MegatronBertModel
|
46 |
|
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)
|