Update README.md

README.md

@@ -22,7 +22,7 @@ Note that this model is primarily aimed at being fine-tuned on math-related task
 Here is how to use this model to get the features of a given text in PyTorch:
 
 ```from transformers import BertTokenizer, BertModel
-tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
+tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT'，output_hidden_states=True)
 model = BertModel.from_pretrained("tbs17/MathBERT")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
@@ -32,7 +32,7 @@ and in TensorFlow:
 
 ```
 from transformers import BertTokenizer, TFBertModel
-tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
+tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT',output_hidden_states=True)
 model = TFBertModel.from_pretrained("tbs17/MathBERT")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='tf')