import gradio as gr
import sambanova_gradio

with gr.Blocks() as demo:
    with gr.Tab("1B"):
        gr.Load("Meta-Llama-3.2-1B-lnstruct", src=sambanova_gradio.registry)
    with gr.Tab("3B"):
        gr.load("Meta-Llama-3.2-3B-lnstruct", src=sambanova_gradio.registry)
    with gr.Tab("8B"):
        gr.load("Meta-Llama-3.1-8B-Instruct", src=sambanova_gradio.registry)
    with gr.Tab("70B"):
        gr.load('Meta-Llama-3.1-70B-Instruct-8k', src=sambanova_gradio.registry)
    with gr.Tab("405B"):
        gr.load('Meta-Llama-3.1-405B-Instruct', src=sambanova_gradio.registry)
        
demo.launch()