add ms florence

This commit is contained in:
Vladimir Mandic
2024-06-23 08:50:06 -04:00
parent 61d4acda68
commit 72f6cf6a1f
5 changed files with 49 additions and 4 deletions
+1 -1
View File
@@ -80,7 +80,7 @@ def create_ui():
with gr.Row():
vqa_answer = gr.Textbox(label="Answer", lines=3)
with gr.Row(elem_id='interrogate_buttons_query'):
vqa_model = gr.Dropdown(list(vqa.MODELS), value='Moondream 2', label='VQA Model')
vqa_model = gr.Dropdown(list(vqa.MODELS), value='MS Florence 2 Base', label='VQA Model')
vqa_submit = gr.Button("Interrogate", elem_id="interrogate_btn_interrogate", variant='primary')
vqa_submit.click(vqa.interrogate, inputs=[vqa_question, vqa_image, vqa_model], outputs=[vqa_answer])