def XlitEngine( lang2use = "all", beam_width=4, rescore=True, model_type = "transformer", src_script_type = "roman", ): if model_type == "transformer": if src_script_type in {"roman", "latin", "en"}: from .transformer import XlitEngineTransformer_En2Indic return XlitEngineTransformer_En2Indic(lang2use, beam_width=beam_width, rescore=rescore) elif src_script_type == "indic": from .transformer import XlitEngineTransformer_Indic2En return XlitEngineTransformer_Indic2En(beam_width=beam_width, rescore=rescore) elif model_type == "rnn": assert src_script_type in {"roman", "latin", "en"} from .rnn.engine import XlitEngineRNN return XlitEngineRNN(lang2use, beam_width=beam_width, rescore=rescore) raise NotImplementedError()