def get_settings(model, beam):
"""
Initialize and customize settings.
"""
translation_settings = TranslationSettings()
translation_settings.models = [model]
translation_settings.num_processes = 1
translation_settings.beam_width = beam
translation_settings.normalization_alpha = 1.0
translation_settings.suppress_unk = True
translation_settings.get_word_probs = False
return translation_settings
def get_settings():
"""
Initialize and customize settings.
"""
translation_settings = TranslationSettings()
translation_settings.models = ["models/model.npz"]
translation_settings.num_processes = 1
translation_settings.beam_width = 12
translation_settings.normalization_alpha = 1.0
translation_settings.suppress_unk = True
translation_settings.get_word_probs = True
return translation_settings
