import os
import gradio as gr

# Load FLUX.1-schnell from Hugging Face Hub as a serverless GPU-backed interface!
# gr.load automatically uses the HF_TOKEN environment variable if you set it in your Space Settings.
# It runs generations on HF's high-speed GPU infrastructure for free in 2-4 seconds!
demo = gr.load("models/black-forest-labs/FLUX.1-schnell")

if __name__ == "__main__":
    demo.launch()