Update app.py

app.py

@@ -51,7 +51,7 @@ class Detect():
         from huggingface_hub import hf_hub_download
         model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="model.bin")
         model = fasttext.load_model(model_path)
-        language, probabilities = model.predict(self.text)
+        language, probabilities = model.predict(self.text, k=3)
         reversed_nllb_langs = {v: k for k, v in languagecodes.nllb_language_codes.items()}
         long_langname = reversed_nllb_langs[language[0].replace('__label__', '')]
         lang_code = all_langs[long_langname][0]
@@ -61,7 +61,7 @@ class Detect():
         from huggingface_hub import hf_hub_download
         model_path = hf_hub_download(repo_id="laurievb/OpenLID-v2", filename="model.bin")
         model = fasttext.load_model(model_path)
-        language, probabilities = model.predict(self.text, , k=3)
+        language, probabilities = model.predict(self.text, k=3)
         reversed_nllb_langs = {v: k for k, v in languagecodes.nllb_language_codes.items()}
         long_langname = reversed_nllb_langs[language[0].replace('__label__', '')]
         lang_code = all_langs[long_langname][0]