fragata commited on
Commit
c2582ae
1 Parent(s): 4700a34

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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-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)
 
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)