Create README.md

README.md

@@ -0,0 +1,33 @@
+---
+language_details: "quy_Latn, spa_Latn"
+pipeline_tag: translation
+tags:
+- nllb
+license: "cc-by-nc-4.0"
+inference: false
+---
+
+# Description
+Finetuned [facebook/nllb-200-3.3B](https://huggingface.co/facebook/nllb-200-3.3B) model to translate between Spanish ("spa_Latn") and Mapuzungun ("quy_Latn").
+
+# Example
+```python
+from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = NllbTokenizer.from_pretrained("CenIA/nllb-200-3.3B-spa-arn", use_auth_token="HF_TOKEN")
+model = AutoModelForSeq2SeqLM.from_pretrained("CenIA/nllb-200-3.3B-spa-arn", use_auth_token="HF_TOKEN")
+
+def translate(sentence: str, translate_from="spa_Latn", translate_to="quy_Latn") -> str:
+    tokenizer.src_lang = translate_from
+    tokenizer.tgt_lang = translate_to
+
+    inputs = tokenizer(sentence, return_tensors="pt")
+    result = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[translate_to])
+    decoded = tokenizer.batch_decode(result, skip_special_tokens=True)[0]
+
+    return decoded
+
+traduction = translate("Hola, ¿cómo estás?")
+
+print(traduction)
+```