fragata commited on
Commit
8d241b4
1 Parent(s): 092990a

Update README.md

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