Update: Make language check softer

app.py

@@ -100,19 +100,15 @@ def _predict_np2_compat(self, text, k=1, threshold=0.0, on_unicode_error='strict
 lid_model.predict = types.MethodType(_predict_np2_compat, lid_model)
 
 ### Check if lang is english #####################################################
-
 def is_eng(review: str):
-    lang_labels, lang_probs = lid_model.predict(review)
+    lang_labels, lang_probs = lid_model.predict(review, k=3)
 
-    print(lang_labels, lang_probs)
+    if "__label__en" in lang_labels:
+        idx = lang_labels.index("__label__en")
 
-    if not lang_labels:  # kein Label zurückgegeben
-        return False, 0.0
+        return True, float(lang_probs[idx])
     
-    lang_label = lang_labels[0]
-    lang_prob = float(lang_probs[0])
-
-    return lang_label[1] == "__label__en", lang_label[1], lang_prob
+    return False, 0.0
 
 ### Do actual prediction #########################################################
 @spaces.GPU(duration=10)  # Sekunden GPU-Zeit pro Call
@@ -126,13 +122,13 @@ def predict(review: str):
         return "Please enter a review.", {}
     
     # Check for lang of text
-    review_is_eng, review_lang_label, review_lang_prob = is_eng(review)
+    review_is_eng, review_lang_prob = is_eng(review)
 
     # Abort if text is not english
     if not review_is_eng:
         # immer zwei Outputs zurückgeben
         return "Only English reviews are supported.", {
-            "lang": review_lang_label.replace("__label__", ""),
+            "is_eng": review_is_eng,
             "prob": review_lang_prob
         }