BrainAI-1 commited on
Commit
23f926c
·
verified ·
1 Parent(s): a385858

Update utils/emotion_detection_brainai.py

Browse files
Files changed (1) hide show
  1. utils/emotion_detection_brainai.py +2 -2
utils/emotion_detection_brainai.py CHANGED
@@ -94,8 +94,8 @@ class EmotionModel:
94
  uploaded_img = PIL.Image.open(img)
95
  uploaded_img_cv = np.array(uploaded_img)
96
 
97
- boxes = detect_faces(uploaded_img_cv, conf = 0.5)
98
- detect_emotions(uploaded_img_cv)
99
 
100
  return uploaded_img_cv
101
 
 
94
  uploaded_img = PIL.Image.open(img)
95
  uploaded_img_cv = np.array(uploaded_img)
96
 
97
+ boxes = self.detect_faces(uploaded_img_cv, conf = 0.5)
98
+ self.detect_emotions(uploaded_img_cv, boxes)
99
 
100
  return uploaded_img_cv
101