Fix example code

README.md

@@ -484,11 +484,15 @@ Find below some example scripts on how to use the model:
 <details>
 <summary> Click to expand </summary>
 
+First, install the Python packages that are required:
+
+`pip install transformers accelerate sentencepiece`
+
 ```python
-from transformers import T5ForConditionalGeneration, T5Tokenizer, GenerationConfig
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 
 model_name = 'jbochi/madlad400-3b-mt'
-model = T5ForConditionalGeneration.from_pretrained(model_name, device="auto")
+model = T5ForConditionalGeneration.from_pretrained(model_name, device_map="auto")
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 
 text = "<2pt> I love pizza!"