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()