This package allows you to obtain translations from Masakhane JoeyNMT based models. Masakhane is a grassroots research community aiming to revive and strengthen African languages through AI.
Project description
AfroTranslate
This package allows you to obtain translations from Masakhane JoeyNMT based models with very few lines of code. Masakhane is a grassroots research community aiming to revive and strengthen African languages through AI.
Available models can be found here.
Note: Please, install the cuda supported version of pytorch to use the GPU. Ex: pip install torch==1.8.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html.
Here are some useful links:
Masakhane: Visit the Masakhane home page.
Masakhane web: Get translations using the Masakhane web application.
Masakhane Github: Take a look at the community main projects here!
JoeyNMT Github: Get hands on JoeyNMT here!
Test it right now on Colab!
Installation
pip install AfroTranslate
Example:
Translation using the English to Fon model
from afrotranslate import MasakhaneTranslate
translator = MasakhaneTranslate(model_name="en-fon")
translator.translate("I love you so much!", n_best=1)
'Un yí wǎn nú we tawun'
Loading a model from specified directory
translator = MasakhaneTranslate(model_path="<directory-where-your-model-resides>")
Translating several sentences at once
The models are trained on individual sentences, so we automatically detect sentence boundaries in inputs and translate them separately. The output shows alternatives for each of them.
translator.translate("I love you so much! Our love is very strong!", n_best=1)
{'Sentence1': 'Un yí wǎn nú we tawun',
'Sentence2': 'Wanyiyi mǐtɔn ɖò taji tawun'}
Disclaimer
This is a community research project and as such, this service is not a production system. The models are only trained using religious data. Therefore, it should not be used for official translations.
Acknowledgement
I want to thank Julia Kreutzer for her precious feedback on this work.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for AfroTranslate-0.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 81fd54cb2de0caa45dcc4fbbd4b2d4f7896c964aeda2d782a3d16c63e04db3a7 |
|
MD5 | 2113fc1c4e15704d80d18485af5c6eb9 |
|
BLAKE2b-256 | 61e909c17dc074dc72f8d654e4581592e775773ff53c6a72d42e1a2e793eb54b |