import gradio as gr
from text2image import text2image_gr
from image2image import image2text_gr

if __name__ == "__main__":
    gr.close_all()
    with gr.TabbedInterface(
            [text2image_gr(), image2text_gr()],
            ["文到图搜索", "图到图搜索"],
    ) as demo:
        demo.launch(share=True)