Update README.md

README.md

@@ -31,7 +31,7 @@ The results on test set.
 ``` bash
 import torch
 
-from transformers import MT5ForConditionalGeneration, T5Tokenizer
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 import torch
 if torch.cuda.is_available():       
     device = torch.device("cuda")
@@ -43,7 +43,7 @@ else:
     print('No GPU available, using the CPU instead.')
     device = torch.device("cpu")
 
-model = MT5ForConditionalGeneration.from_pretrained("NlpHUST/t5-en-vi-small")
+model = T5ForConditionalGeneration.from_pretrained("NlpHUST/t5-en-vi-small")
 tokenizer = T5Tokenizer.from_pretrained("NlpHUST/t5-en-vi-small")
 model.to(device)