Update README.md

README.md

@@ -46,8 +46,8 @@ 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-custom')
-model = BertModel.from_pretrained("tbs17/MathBERT-custom")
+tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
+model = BertModel.from_pretrained("tbs17/MathBERT")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(encoded_input)
@@ -56,8 +56,8 @@ and in TensorFlow:
 
 ```
 from transformers import BertTokenizer, TFBertModel
-tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT-custom')
-model = TFBertModel.from_pretrained("tbs17/MathBERT-custom")
+tokenizer = BertTokenizer.from_pretrained('tbs17/MathBERT')
+model = TFBertModel.from_pretrained("tbs17/MathBERT")
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='tf')
 output = model(encoded_input)