import gradio as gr
from inference import generate_image

# Gradio UI
demo = gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(placeholder="Describe your image..."),
    outputs="image",
    title="Text to Image Generator",
    description="Powered by cloudqi/cqi_text_to_image_pt_v0"
)

demo.launch(share=True)