diff --git a/opus_mt.py b/opus_mt.py index 9860433..5dd8847 100644 --- a/opus_mt.py +++ b/opus_mt.py @@ -52,4 +52,16 @@ class OpusMT(NNOperator): configs['opus-mt-zh-en'] = {} configs['opus-mt-zh-en']['tokenizer'] = 'Helsinki-NLP/opus-mt-zh-en' configs['opus-mt-zh-en']['model'] = 'Helsinki-NLP/opus-mt-zh-en' + + configs['opus-mt-zh-en'] = {} + configs['opus-mt-zh-en']['tokenizer'] = 'Helsinki-NLP/opus-mt-zh-en' + configs['opus-mt-zh-en']['model'] = 'Helsinki-NLP/opus-mt-zh-en' + + configs['opus-tatoeba-en-ja'] = {} + configs['opus-tatoeba-en-ja']['tokenizer'] = 'Helsinki-NLP/opus-tatoeba-en-ja' + configs['opus-tatoeba-en-ja']['model'] = 'Helsinki-NLP/opus-tatoeba-en-ja' + + configs['opus-tatoeba-ja-en'] = {} + configs['opus-tatoeba-ja-en']['tokenizer'] = 'Helsinki-NLP/opus-tatoeba-ja-en' + configs['opus-tatoeba-ja-en']['model'] = 'Helsinki-NLP/opus-tatoeba-ja-en' return configs