Update README.md

README.md

@@ -1,5 +1,4 @@
-
-## BLEURT
+\n## BLEURT
 
 Pytorch version of the original BLEURT models from ACL paper ["BLEURT: Learning Robust Metrics for Text Generation"](https://aclanthology.org/2020.acl-main.704/) by 
 Thibault Sellam, Dipanjan Das and Ankur P. Parikh of Google Research.
@@ -12,8 +11,8 @@ The code for model conversion was originated from [this notebook](https://colab.
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("elron/bleurt-large-512")
-model = AutoModelForSequenceClassification.from_pretrained("elron/bleurt-large-512")
+tokenizer = AutoTokenizer.from_pretrained("Elron/bleurt-base-512")
+model = AutoModelForSequenceClassification.from_pretrained("Elron/bleurt-base-512")
 model.eval()
 
 references = ["hello world", "hello world"]