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

silly if fix in vqa

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-23 11:57:03 -04:00
parent 3e6282b7b6
commit 50de0c1943
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/vqa.py
+5 -5
View File
@@ -178,15 +178,15 @@ def interrogate(vqa_question, vqa_image, vqa_model_req):
return answer
if 'git' in vqa_model.lower():
answer = git(vqa_question, vqa_image, vqa_model)
if 'vilt' in vqa_model.lower():
elif 'vilt' in vqa_model.lower():
answer = vilt(vqa_question, vqa_image, vqa_model)
if 'blip' in vqa_model.lower():
elif 'blip' in vqa_model.lower():
answer = blip(vqa_question, vqa_image, vqa_model)
if 'pix' in vqa_model.lower():
elif 'pix' in vqa_model.lower():
answer = pix(vqa_question, vqa_image, vqa_model)
if 'moondream2' in vqa_model.lower():
elif 'moondream2' in vqa_model.lower():
answer = moondream(vqa_question, vqa_image, vqa_model)
if 'florence' in vqa_model.lower():
elif 'florence' in vqa_model.lower():
answer = florence(vqa_question, vqa_image, vqa_model)
else:
answer = 'unknown model'