Spaces:

jaimin
/

Heartandstress

Sleeping

jaimin commited on Oct 11

Commit

d04c369

•

1 Parent(s): f758bba

Update eyebrow_detection_modified_copy.py

Files changed (1) hide show

eyebrow_detection_modified_copy.py CHANGED Viewed

@@ -38,8 +38,8 @@ def normalize_values(points, disp):
 def stress(video_path, duration):
     global points, emotion_classifier
     detector = dlib.get_frontal_face_detector()
-    predictor = dlib.shape_predictor("stress_detection/models/data")
-    emotion_classifier = load_model("stress_detection/models/_mini_XCEPTION.102-0.66.hdf5", compile=False)
     cap = cv2.VideoCapture(video_path)
     points = []

 def stress(video_path, duration):
     global points, emotion_classifier
     detector = dlib.get_frontal_face_detector()
+    predictor = dlib.shape_predictor("models/data")
+    emotion_classifier = load_model("models/_mini_XCEPTION.102-0.66.hdf5", compile=False)
     cap = cv2.VideoCapture(video_path)
     points = []