Views
No views yet
1from transformers import MarianMTModel, MarianTokenizer
2
3src_text = [
4 "La María és feminista.",
5 "Contribuyan en Tatoeba."
6]
7
8model_name = "pytorch-models/opus-mt-tc-big-itc-he"
9tokenizer = MarianTokenizer.from_pretrained(model_name)
10model = MarianMTModel.from_pretrained(model_name)
11translated = model.generate(**tokenizer(src_text, return_tensors="pt", padding=True))
12
13for t in translated:
14 print( tokenizer.decode(t, skip_special_tokens=True) )
15
16# expected output:
17# מרי היא פמיניסטית.
18# תרום לטאטואבה.1from transformers import pipeline
2pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-itc-he")
3print(pipe("La María és feminista."))
4
5# expected output: מרי היא פמיניסטית.| langpair | testset | chr-F | BLEU | #sent | #words |
|---|---|---|---|---|---|
| fra-heb | tatoeba-test-v2021-08-07 | 0.60539 | 39.6 | 3281 | 20655 |
| ita-heb | tatoeba-test-v2021-08-07 | 0.60264 | 40.0 | 1706 | 9796 |
| por-heb | tatoeba-test-v2021-08-07 | 0.63087 | 44.4 | 719 | 4423 |
| spa-heb | tatoeba-test-v2021-08-07 | 0.63883 | 44.5 | 1849 | 12112 |
| cat-heb | flores101-devtest | 0.52457 | 23.0 | 1012 | 20749 |
| fra-heb | flores101-devtest | 0.52953 | 23.2 | 1012 | 20749 |
| glg-heb | flores101-devtest | 0.50918 | 20.8 | 1012 | 20749 |
| ita-heb | flores101-devtest | 0.49007 | 18.3 | 1012 | 20749 |
| por-heb | flores101-devtest | 0.53906 | 24.4 | 1012 | 20749 |
| ron-heb | flores101-devtest | 0.52103 | 22.1 | 1012 | 20749 |
| spa-heb | flores101-devtest | 0.47646 | 16.5 | 1012 | 20749 |
@inproceedings{tiedemann-thottingal-2020-opus,
title = "{OPUS}-{MT} {--} Building open translation services for the World",
author = {Tiedemann, J{\"o}rg and Thottingal, Santhosh},
booktitle = "Proceedings of the 22nd Annual Conference of the European Association for Machine Translation",
month = nov,
year = "2020",
address = "Lisboa, Portugal",
publisher = "European Association for Machine Translation",
url = "https://aclanthology.org/2020.eamt-1.61",
pages = "479--480",
}
@inproceedings{tiedemann-2020-tatoeba,
title = "The Tatoeba Translation Challenge {--} Realistic Data Sets for Low Resource and Multilingual {MT}",
author = {Tiedemann, J{\"o}rg},
booktitle = "Proceedings of the Fifth Conference on Machine Translation",
month = nov,
year = "2020",
address = "Online",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2020.wmt-1.139",
pages = "1174--1182",
}