Used fasttext for language detection

app.py

@@ -36,7 +36,7 @@ classifiers = {'en': pipeline("zero-shot-classification", hypothesis_template=hy
                'no': pipeline("zero-shot-classification", hypothesis_template=hypothesis_templates['no'],
                               model=models['no'])}
 
-fasttext_model = fasttext.load_model(hf_hub_download("julien-c/fasttext-language-id", "model.bin"))
+fasttext_model = fasttext.load_model(hf_hub_download("julien-c/fasttext-language-id", "lid.176.bin"))
 
 def prep_examples():
     example_text1 = "Coronavirus disease (COVID-19) is an infectious disease caused by the SARS-CoV-2 virus. Most \
@@ -104,10 +104,10 @@ def detect_lang(sequence, labels):
     seq_lang = 'en'
 
     try:
-        # seq_lang = detect(sequence)
-        # lbl_lang = detect(labels)
-        seq_lang = fasttext_model.predict(sequence)[0][0].split("__label__")[1]
-        lbl_lang = fasttext_model.predict(labels)[0][0].split("__label__")[1]
+        #seq_lang = detect(sequence)
+        #lbl_lang = detect(labels)
+        seq_lang = fasttext_model.predict(sequence, k=1)[0][0].split("__label__")[1]
+        lbl_lang = fasttext_model.predict(labels, k=1)[0][0].split("__label__")[1]
     except:
         print("Language detection failed!",
               "Date:{}, Sequence:{}, Labels:{}".format(