import gradio as gr import requests import io import random import os from PIL import Image API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl" API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free headers = {"Authorization": f"Bearer {API_TOKEN}"} def select_model(model_name): models_list = [ "openskyml/dalle-3-xl", "Linaqruf/animagine-xl-2.0", "Lykon/dreamshaper-7", "Linaqruf/animagine-xl", "runwayml/stable-diffusion-v1-5", "stabilityai/stable-diffusion-xl-base-1.0", "prompthero/openjourney-v4", "nerijs/pixel-art-xl", "Linaqruf/anything-v3.0" ] def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None): payload = { "inputs": prompt, "is_negative": is_negative, "steps": steps, "cfg_scale": cfg_scale, "seed": seed if seed is not None else random.randint(-1, 2147483647) } image_bytes = requests.post(API_URL, headers=headers, json=payload).content image = Image.open(io.BytesIO(image_bytes)) return image with gr.Blocks(theme="soft") as playground: gr.HTML( """
Create your AI art with Stable Diffusion models!