batterydata
commited on
Commit
•
a90bcf9
1
Parent(s):
66a30be
Update README.md
Browse files
README.md
CHANGED
@@ -55,7 +55,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
|
|
55 |
```python
|
56 |
from transformers import BertTokenizer, BertModel
|
57 |
tokenizer = BertTokenizer.from_pretrained('batterydata/batteryonlybert-uncased')
|
58 |
-
model = BertModel.from_pretrained(
|
59 |
text = "Replace me by any text you'd like."
|
60 |
encoded_input = tokenizer(text, return_tensors='pt')
|
61 |
output = model(**encoded_input)
|
@@ -66,7 +66,7 @@ and in TensorFlow:
|
|
66 |
```python
|
67 |
from transformers import BertTokenizer, TFBertModel
|
68 |
tokenizer = BertTokenizer.from_pretrained('batterydata/batteryonlybert-uncased')
|
69 |
-
model = TFBertModel.from_pretrained(batterydata/batteryonlybert-uncased
|
70 |
text = "Replace me by any text you'd like."
|
71 |
encoded_input = tokenizer(text, return_tensors='tf')
|
72 |
output = model(encoded_input)
|
|
|
55 |
```python
|
56 |
from transformers import BertTokenizer, BertModel
|
57 |
tokenizer = BertTokenizer.from_pretrained('batterydata/batteryonlybert-uncased')
|
58 |
+
model = BertModel.from_pretrained('batterydata/batteryonlybert-uncased')
|
59 |
text = "Replace me by any text you'd like."
|
60 |
encoded_input = tokenizer(text, return_tensors='pt')
|
61 |
output = model(**encoded_input)
|
|
|
66 |
```python
|
67 |
from transformers import BertTokenizer, TFBertModel
|
68 |
tokenizer = BertTokenizer.from_pretrained('batterydata/batteryonlybert-uncased')
|
69 |
+
model = TFBertModel.from_pretrained('batterydata/batteryonlybert-uncased')
|
70 |
text = "Replace me by any text you'd like."
|
71 |
encoded_input = tokenizer(text, return_tensors='tf')
|
72 |
output = model(encoded_input)
|