Update README.md

README.md

@@ -1,3 +1,4 @@
+
 ## 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 
@@ -8,7 +9,7 @@ The code for model conversion was originated from [this notebook](https://colab.
 ## Usage Example
 
 ```python
-from transformers import BertForSequenceClassification, AutoTokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
 tokenizer = AutoTokenizer.from_pretrained("elron/bleurt-large-512")
@@ -22,4 +23,4 @@ with torch.no_grad():
  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0]
 
 print(scores) # tensor([[0.9877], [0.0475]])
-```
+```