gchhablani commited on
Commit
fb33b8e
1 Parent(s): 9dfc631

Update README

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ generation you should look at model like GPT2.
47
  Here is how to use this model to get the features of a given text in PyTorch:
48
  ```python
49
  from transformers import BertTokenizer, BertModel
50
- tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
51
  model = BertModel.from_pretrained("multiberts-seed-1-600k")
52
  text = "Replace me by any text you'd like."
53
  encoded_input = tokenizer(text, return_tensors='pt')
 
47
  Here is how to use this model to get the features of a given text in PyTorch:
48
  ```python
49
  from transformers import BertTokenizer, BertModel
50
+ tokenizer = BertTokenizer.from_pretrained('multiberts-seed-1-600k')
51
  model = BertModel.from_pretrained("multiberts-seed-1-600k")
52
  text = "Replace me by any text you'd like."
53
  encoded_input = tokenizer(text, return_tensors='pt')