batterydata commited on
Commit
a90bcf9
1 Parent(s): 66a30be

Update README.md

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