import gradio as gr
from huggingface_hub import InferenceClient
import os
from dotenv import load_dotenv

load_dotenv()

model = "stabilityai/stable-diffusion-xl-base-1.0"
client = InferenceClient(model,token = os.getenv("HF_TOKEN"))

def generate_image(prompt):
    return client.text_to_image(prompt)

demo = gr.Interface(
    fn=generate_image,
    inputs = gr.Textbox(label="Enter prompt"),
    outputs = "image",
    title = "AI Image Generator",
    
)


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