ziem-io commited on
Commit
577c02d
·
1 Parent(s): 9ea7979

Fix: Lang check

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -50,10 +50,14 @@ lid_model.predict = types.MethodType(_predict_np2_compat, lid_model)
50
 
51
  def is_eng(review: str):
52
  lang_labels, lang_probs = lid_model.predict(review)
 
 
 
 
53
  lang_label = lang_labels[0]
54
  lang_prob = float(lang_probs[0])
55
 
56
- return lang_label[1] == "__lang_en__", lang_prob
57
 
58
  ### Do actual prediction ##############################################
59
 
 
50
 
51
  def is_eng(review: str):
52
  lang_labels, lang_probs = lid_model.predict(review)
53
+
54
+ if not lang_labels: # kein Label zurückgegeben
55
+ return False, 0.0
56
+
57
  lang_label = lang_labels[0]
58
  lang_prob = float(lang_probs[0])
59
 
60
+ return lang_label[1] == "__label__en", lang_prob
61
 
62
  ### Do actual prediction ##############################################
63