File size: 952 Bytes
87a18d2 f6d9d47 3a0f530 f6d9d47 6641975 0fabc81 f6d9d47 85cf7d8 f6d9d47 |
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 34 |
---
language:
- ru
- bua
- bxr
datasets:
- SaranaAbidueva/buryat-russian_parallel_corpus
metrics:
- bleu
---
This model translated from Russian to Buryat language.
How to use in Python:
```python
from transformers import MBartForConditionalGeneration, MBart50Tokenizer
model = MBartForConditionalGeneration.from_pretrained("SaranaAbidueva/mbart50_ru_bua")
tokenizer = MBart50Tokenizer.from_pretrained("SaranaAbidueva/mbart50_ru_bua")
def translate(text, max_length=200, num_beams=5, repetition_penalty=5.0, **kwargs):
encoded = tokenizer(text, return_tensors="pt")
generated_tokens = model.generate(
**encoded.to(model.device),
max_length=max_length,
num_beams=num_beams,
repetition_penalty=repetition_penalty,
# early_stopping=True,
)
return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
translate('Евгений Онегин интересная книга')
``` |