import gradio as gr

with gr.Blocks(fill_height=True) as demo:
    with gr.Sidebar():
        gr.Markdown("# DiaGen")
        gr.Markdown("This Space showcases the DiaGen model. Sign in with your Hugging Face account to use this API.")
        button = gr.LoginButton("Sign in")
    gr.load("models/nari-labs/Dia-1.6B", accept_token=button, provider="fal-ai")
    
demo.launch()