import gradio as gr

with gr.Blocks(fill_height=True) as demo:
    with gr.Sidebar():
        gr.Markdown("# Inference Provider")
        gr.Markdown("This Space showcases the THUDM/CogView4-6B model, served by the fal-ai API. Sign in with your Hugging Face account to use this API.")
        button = gr.LoginButton("Sign in")
    gr.load("models/THUDM/CogView4-6B", accept_token=button, provider="fal-ai")
    
demo.launch()