Update README.md

README.md

@@ -2,6 +2,10 @@
 license: apache-2.0
 language:
 - en
+- pt
+- es
+- de
+- it
 library_name: transformers
 tags:
 - text-generation-inference
@@ -9,7 +13,26 @@ tags:
 
 T5ForConditionalGeneration files for the [Madlad-400](https://github.com/google-research/google-research/tree/master/madlad_400) 3B parameter MT model.
 
-**This is a WIP. I'm able to load and run the model, but output is nonsensical.**
 
-Colab to generate these files is [here](https://colab.research.google.com/drive/1rZ2NRyl2zwmg0sQ2Wi-uZZF48iVYulTC#scrollTo=pVODoE6gA9sw).
+```python
+from transformers import T5ForConditionalGeneration, T5Tokenizer, GenerationConfig
 
+model = T5ForConditionalGeneration.from_pretrained('jbochi/madlad400-3b-mt')
+
+tokenizer = T5Tokenizer.from_pretrained('./madlad400-3b-mt')
+
+text = "<2es> how do you say torch in portuguese?"
+input_ids = tokenizer(text, return_tensors="pt").input_ids
+
+outputs = model.generate(
+    input_ids=input_ids,
+    generation_config=GenerationConfig(
+        decoder_start_token_id=2,
+))
+
+tokenizer.decode(outputs[0], skip_special_tokens=True)
+# como se dice antorcha en portugués?
+```
+
+
+Colab to generate these files is [here](https://colab.research.google.com/drive/1rZ2NRyl2zwmg0sQ2Wi-uZZF48iVYulTC#scrollTo=pVODoE6gA9sw).