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.responses import JSONResponse
from blackbox import Blackbox
from .blackbox import Blackbox
from typing import Optional
import requests
@ -48,14 +48,17 @@ class VLMS(Blackbox):
return data.text
async def fast_api_handler(self, request: Request) -> Response:
try:
data = await request.json()
except:
return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
# try:
# data = await request.json()
# except:
# return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
model_name = data.get("model_name")
prompt = data.get("prompt")
img_data = data.get("img_data")
# model_name = data.get("model_name")
# prompt = data.get("prompt")
# 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:
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():
model_name = "Qwen-VL-Chat"
return JSONResponse(content={"response": self.processing(model_name, prompt, img_data)},
status_code=status.HTTP_200_OK)
jsonresp = str(JSONResponse(content={"response": self.processing(prompt, img_data, model_name)}).body, "utf-8")
return JSONResponse(content={"response": jsonresp}, status_code=status.HTTP_200_OK)