Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): update VQA API endpoint for tuple return format

Browse Source 
Update interrogate API endpoint to handle the new (text, image)
tuple return format from VQA interrogate function.
This commit is contained in:
CalamitousFelicitousness
2025-12-04 00:05:10 +00:00
parent 27fa48cc99
commit c2810dfee2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/api/endpoints.py
+1 -1
View File
@@ -100,7 +100,7 @@ def post_vqa(req: models.ReqVQA):
image = helpers.decode_base64_to_image(req.image)
image = image.convert('RGB')
from modules.interrogate import vqa
answer = vqa.interrogate(req.question, req.system, '', image, req.model)
answer, _ = vqa.interrogate(req.question, req.system, '', image, req.model)
return models.ResVQA(answer=answer)
def post_unload_checkpoint():