Mohammedelhakim commited on
Commit
93a1b57
·
verified ·
1 Parent(s): 744ff77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def detect_is_cry(path, threshold):
97
  feat = det_scaler.transform(feat)
98
  feat = det_pca.transform(feat)
99
 
100
- prob = det_model.predict_proba(feat)[0][1]
101
 
102
  is_cry = bool(prob >= threshold)
103
  return is_cry, float(prob)
 
97
  feat = det_scaler.transform(feat)
98
  feat = det_pca.transform(feat)
99
 
100
+ prob = det_model.predict_proba(feat)[0][0]
101
 
102
  is_cry = bool(prob >= threshold)
103
  return is_cry, float(prob)