Update app.py
Browse files
app.py
CHANGED
|
@@ -99,7 +99,7 @@ def detect_is_cry(path, threshold):
|
|
| 99 |
|
| 100 |
prob = det_model.predict_proba(feat)[0][0]
|
| 101 |
|
| 102 |
-
is_cry = bool(prob
|
| 103 |
return is_cry, float(prob)
|
| 104 |
|
| 105 |
|
|
|
|
| 99 |
|
| 100 |
prob = det_model.predict_proba(feat)[0][0]
|
| 101 |
|
| 102 |
+
is_cry = bool(prob <= threshold)
|
| 103 |
return is_cry, float(prob)
|
| 104 |
|
| 105 |
|