From 50de0c19438425fa6a3e8e2db4a443c02eea2fa6 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 23 Jun 2024 11:57:03 -0400 Subject: [PATCH] silly if fix in vqa --- modules/vqa.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/modules/vqa.py b/modules/vqa.py index 30e0031ae..5a17c091e 100644 --- a/modules/vqa.py +++ b/modules/vqa.py @@ -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'