import gradio as gr
import sambanova_gradio

with gr.Blocks(fill_height=True,theme=gr.themes.Ocean()) as demo:
    gr.load(
        name='Meta-Llama-3.1-405B-Instruct',
        src=sambanova_gradio.registry,
        title='Sambanova-Gradio Integration',
        description="Chat with Meta-Llama-3.1-405B-Instruct model.",
        examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."]
    )

demo.launch()