Commit
•
1609e3e
1
Parent(s):
eab127d
Updated bug in TensorFlow usage code (README.md) (#5)
Browse files- Updated bug in TensorFlow usage code (README.md) (8517e88f3753188f27fefe1ee6b52b48c9bbbd37)
Co-authored-by: Pranshu Pant <[email protected]>
README.md
CHANGED
|
@@ -114,7 +114,7 @@ and in TensorFlow:
|
|
| 114 |
```python
|
| 115 |
from transformers import AlbertTokenizer, TFAlbertModel
|
| 116 |
tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
|
| 117 |
-
model = TFAlbertModel.from_pretrained("albert-base-v2)
|
| 118 |
text = "Replace me by any text you'd like."
|
| 119 |
encoded_input = tokenizer(text, return_tensors='tf')
|
| 120 |
output = model(encoded_input)
|
|
|
|
| 114 |
```python
|
| 115 |
from transformers import AlbertTokenizer, TFAlbertModel
|
| 116 |
tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
|
| 117 |
+
model = TFAlbertModel.from_pretrained("albert-base-v2")
|
| 118 |
text = "Replace me by any text you'd like."
|
| 119 |
encoded_input = tokenizer(text, return_tensors='tf')
|
| 120 |
output = model(encoded_input)
|