No caso da seleção da linga poderia ser utilizado o argmin da bilbioteca numpy
import numpy as np
langs = ["pt","en","es"]
models = {
"pt": model_pt,
"en": model_en,
"es": model_es,
}
def classify(text, langs):
perplexities = []
for lang in langs:
perplexities.append(perplexity(models[lang],text))
return langs[np.argmin(perplexities)]