Spaces:

nsa
/

Facial-Emotion-Recognition

Sleeping

NyanSwanAung23 commited on Oct 18, 2023

Commit

d0eefe5

1 Parent(s): edb429d

change labels

Files changed (1) hide show

classifier.py CHANGED Viewed

@@ -13,7 +13,7 @@ class Classifier:
         self.classifier = None
         self.face_detector = None
         self.labels = {0: "Angry", 1: "Disgusted", 2: "Fearful",
-                       3: "Happy", 4: "Neutral", 5: "Sad", 6: "Surprised"}
         self.load_models()
     def check_file_type(self, is_image, input_path):

         self.classifier = None
         self.face_detector = None
         self.labels = {0: "Angry", 1: "Disgusted", 2: "Fearful",
+                       3: "Happy", 4: "Neutral", 5: "Surprised", 6: "Sad"}
         self.load_models()
     def check_file_type(self, is_image, input_path):