import gradio as gr | |
from image_captioning import get_image_captioning_tab | |
from visual_qa.py import get_visual_qa_tab | |
from zero_shot_classification import get_zero_shot_classification_tab | |
with gr.Blocks() as app: | |
image_captioning_tab = get_image_captioning_tab() | |
visual_qa_tab = get_visual_qa_tab() | |
zero_shot_classification_tab = get_zero_shot_classification_tab() | |
gr.TabbedInterface( | |
[image_captioning_tab, visual_qa_tab, zero_shot_classification_tab], | |
["Image Captioning", "Visual Q&A", "Zero-Shot Classification"] | |
) | |
app.launch() |