| 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() |