Update README.md

README.md

@@ -20,9 +20,9 @@ To force the target language id as the first generated token, pass the `forced_b
 To install `sentencepiece` run `pip install sentencepiece`
 
 ```python
-from transformers import AutoModelForConditionalGeneration, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-model = AutoModelForConditionalGeneration.from_pretrained("facebook/wmt21-dense-24-wide-en-x")
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/wmt21-dense-24-wide-en-x")
 tokenizer = AutoTokenizer.from_pretrained("facebook/wmt21-dense-24-wide-en-x")
 
 inputs = tokenizer("One model for many languages.", return_tensors="pt")