Hugging Face's logo

tbs17
/
MathBERT

Fill-Mask
Transformers
PyTorch
bert
Inference Endpoints
Model card Files Community
2
tbs17 commited on
Commit
e26235c
·
1 Parent(s): 75866ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -22,7 +22,7 @@ Note that this model is primarily aimed at being fine-tuned on math-related task
22
  Here is how to use this model to get the features of a given text in PyTorch:
23
 
24
  ```from transformers import BertTokenizer, BertModel
25
- tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
26
  model = BertModel.from_pretrained("tbs17/MathBERT")
27
  text = "Replace me by any text you'd like."
28
  encoded_input = tokenizer(text, return_tensors='pt')
@@ -32,7 +32,7 @@ and in TensorFlow:
32
 
33
  ```
34
  from transformers import BertTokenizer, TFBertModel
35
- tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
36
  model = TFBertModel.from_pretrained("tbs17/MathBERT")
37
  text = "Replace me by any text you'd like."
38
  encoded_input = tokenizer(text, return_tensors='tf')
 
22
  Here is how to use this model to get the features of a given text in PyTorch:
23
 
24
  ```from transformers import BertTokenizer, BertModel
25
+ tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT'，output_hidden_states=True)
26
  model = BertModel.from_pretrained("tbs17/MathBERT")
27
  text = "Replace me by any text you'd like."
28
  encoded_input = tokenizer(text, return_tensors='pt')
 
32
 
33
  ```
34
  from transformers import BertTokenizer, TFBertModel
35
+ tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT',output_hidden_states=True)
36
  model = TFBertModel.from_pretrained("tbs17/MathBERT")
37
  text = "Replace me by any text you'd like."
38
  encoded_input = tokenizer(text, return_tensors='tf')