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 17:26:09 +08:00
committed by GitHub
parent 4c01ad46fa
commit 8d8eae158c
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/blackbox/emotion.py
View File

@ -11,7 +11,7 @@ from injector import singleton
@singleton
class Emotion(Blackbox):
def __init__(self, model_name, model_url) -> None:
def __init__(self, model_name:str = "Mistral-7B-Instruct-v0.2", model_url:str = "http://120.196.116.194:48892") -> None:
self.model = LMDeployClient(
model_name=model_name,
url=model_url,
@ -29,10 +29,21 @@ class Emotion(Blackbox):
data = args[0]
return isinstance(data, str)
def processing(self, text: str) -> int:
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}]
return self.model.stream_chat(text)
def processing(self, text: str) -> str:
response = ""
for text in self.model.stream_chat(text):
print("------------")
print(text[1])
print("------------")
# print(type(text))
response = text[1]
# print("************0")
# print(response)
return response
async def fast_api_handler(self, request) -> Response:
try:
@ -45,5 +56,5 @@ class Emotion(Blackbox):
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}]
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)