添加EmotionRecognition模型注释

This commit is contained in:
bing 2024-09-04 17:16:10 +08:00
parent ee3dbb04f0
commit 25b99502fd
1 changed files with 53 additions and 36 deletions

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)