Spaces:
Running
Running
Fix: Lang check
Browse files
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] == "
|
| 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 |
|