utils/emotion2vec_utils.py

@@ -9,6 +9,23 @@ model_args = {
 }
 
 class EmotionRecognition:
+"""
+    9-class emotion
+    0: angry
+    1: disgusted
+    2: fearful
+    3: happy
+    4: neutral
+    5: other
+    6: sad
+    7: surprised
+    8: unknown
+    return :  
+    [{"emotion": "angry", "weight": },
+     {"emotion": "disgusted", "weight": },
+     ...
+    ]
+"""
     def __init__(self) -> None:
         self.initialize(model_args=model_args)