gchhablani
commited on
Commit
•
fb33b8e
1
Parent(s):
9dfc631
Update README
Browse files
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('
|
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')
|