添加EmotionRecognition模型注释

This commit is contained in:
bing committed 2024-09-04 17:16:10 +08:00
1 parent ee3dbb04f0
commit 25b99502fd
1 file changed
+17
+17
View File
@@ -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)