A Machine Translation engine that utilizes a pre-trained model from HuggingFace, specifically the mBART-50 model.
- If you don't have conda installed on your machine, install it by following the instructions at this link:
https://docs.conda.io/projects/conda/en/latest/user-guide/install/ - Create a new environment and install the required packages by running the command:
conda create --name <env> --file requirements.txt
- Set the parameters through config.yaml
- Execute the python script
index.py
The predict.py
script can be utilized to translate sentences with the trained model.