TiberiuCristianLeon commited on
Commit
9418f40
·
verified ·
1 Parent(s): a99d331

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ class Detect():
53
  model = fasttext.load_model(model_path)
54
  language, probabilities = model.predict(self.text)
55
  reversed_nllb_langs = {v: k for k, v in languagecodes.nllb_language_codes.items()}
56
- long_langname = reversed_nllb_langs[language[0].replace('__label__', ''])
57
  lang_code = all_langs[long_langname][0]
58
  return [lang_code, round(number=probabilities[0] * 100, ndigits=2)]
59
 
 
53
  model = fasttext.load_model(model_path)
54
  language, probabilities = model.predict(self.text)
55
  reversed_nllb_langs = {v: k for k, v in languagecodes.nllb_language_codes.items()}
56
+ long_langname = reversed_nllb_langs[language[0].replace('__label__', '')]
57
  lang_code = all_langs[long_langname][0]
58
  return [lang_code, round(number=probabilities[0] * 100, ndigits=2)]
59