import gradio as gr
import os

with gr.Blocks() as demo:
    button = gr.LoginButton("Sign in to use your Hugging Face Token")
    with gr.Column(variant="panel"):
        gr.load("black-forest-labs/FLUX.1-dev", src="models", provider="fal-ai", accept_token=button)
    
demo.launch()

# gr.load("distilbert/distilbert-base-uncased", src="models", accept_token=gr.LoginButton()).launch()