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

vlms updated

Browse Source
This commit is contained in:
gdw6463
2024-05-10 14:32:46 +08:00
parent 739048ef3d
commit d947c917e5
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/blackbox/vlms.py
View File

@ -1,6 +1,6 @@
from fastapi import Request, Response, status from fastapi import Request, Response, status
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from blackbox import Blackbox from .blackbox import Blackbox
from typing import Optional from typing import Optional
import requests import requests
@ -48,14 +48,17 @@ class VLMS(Blackbox):
return data.text return data.text
async def fast_api_handler(self, request: Request) -> Response: async def fast_api_handler(self, request: Request) -> Response:
try: # try:
data = await request.json() # data = await request.json()
except: # except:
return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST) # return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
model_name = data.get("model_name") # model_name = data.get("model_name")
prompt = data.get("prompt") # prompt = data.get("prompt")
img_data = data.get("img_data") # img_data = data.get("img_data")
model_name = request.get("model_name")
prompt = request.get("prompt")
img_data = request.get("img_data")
if prompt is None: if prompt is None:
return JSONResponse(content={'error': "Question is required"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={'error': "Question is required"}, status_code=status.HTTP_400_BAD_REQUEST)
@ -63,5 +66,5 @@ class VLMS(Blackbox):
if model_name is None or model_name.isspace(): if model_name is None or model_name.isspace():
model_name = "Qwen-VL-Chat" model_name = "Qwen-VL-Chat"
return JSONResponse(content={"response": self.processing(model_name, prompt, img_data)}, jsonresp = str(JSONResponse(content={"response": self.processing(prompt, img_data, model_name)}).body, "utf-8")
status_code=status.HTTP_200_OK) return JSONResponse(content={"response": jsonresp}, status_code=status.HTTP_200_OK)