Pytorch example encoded_input fix

README.md

@@ -25,7 +25,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
 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')
+encoded_input = tokenizer(text, return_tensors='pt')["input_ids"]
 output = model(encoded_input)
 ```
 and in TensorFlow: