ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update emotion.py

Browse Source
This commit is contained in:
Limbo
2024-05-13 16:23:32 +08:00
committed by GitHub
parent 27972271a0
commit 4c01ad46fa
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/blackbox/emotion.py
View File

@ -29,8 +29,7 @@ class Emotion(Blackbox):
data = args[0] data = args[0]
return isinstance(data, str) return isinstance(data, str)
def processing(self, *args, **kwargs) -> int: def processing(self, text: str) -> int:
text = args[0]
text = "Please use one word to infer the emotion of the following passage:\n" + text + "\nJust print out that signle word pls." text = "Please use one word to infer the emotion of the following passage:\n" + text + "\nJust print out that signle word pls."
text = [{'role': 'user', 'content': text}] text = [{'role': 'user', 'content': text}]
return self.model.stream_chat(text) return self.model.stream_chat(text)
@ -47,4 +46,4 @@ class Emotion(Blackbox):
text = [{'role': 'user', 'content': text}] text = [{'role': 'user', 'content': text}]
sentiment = self.processing(text) sentiment = self.processing(text)
return JSONResponse(content={"sentiment": sentiment }, status_code=status.HTTP_200_OK) return JSONResponse(content={"sentiment": sentiment }, status_code=status.HTTP_200_OK)