Update app.py
Browse files
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 |
|