import gradio as gr
from huggingface_hub import InferenceClient

# Stable Diffusion XL ka fast model load ho raha hai
client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0")

def generate(prompt):
    # Image generate karega
    image = client.text_to_image(prompt)
    return image

# Gradio Interface
demo = gr.Interface(
    fn=generate, 
    inputs=gr.Textbox(label="Prompt likhein"), 
    outputs=gr.Image(label="Generated Image")
)

demo.launch()