File size: 1,025 Bytes
7e3ad3f 909722d 7e3ad3f 549b544 7e3ad3f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
language_details: "mri_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 Rapanui ("mri_Latn").
# Example
```python
from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
tokenizer = NllbTokenizer.from_pretrained("CenIA/nllb-200-3.3B-spa-rap")
model = AutoModelForSeq2SeqLM.from_pretrained("CenIA/nllb-200-3.3B-spa-rap")
def translate(sentence: str, translate_from="spa_Latn", translate_to="mri_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.convert_tokens_to_ids(translate_to))
decoded = tokenizer.batch_decode(result, skip_special_tokens=True)[0]
return decoded
traduction = translate("Hola, ¿cómo estás?")
print(traduction)
``` |