Update README.md

README.md

@@ -30,7 +30,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")
@@ -42,7 +42,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)