from diffusers import StableDiffusionPipeline import gradio as gr import torch models = [ "DGSpitzer/Cyberpunk-Anime-Diffusion" ] prompt_prefixes = { models[0]: "cyberpunk anime style" } current_model = models[0] pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torch.float16) if torch.cuda.is_available(): pipe = pipe.to("cuda") device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶" def on_model_change(model): global current_model global pipe if model != current_model: current_model = model pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torch.float16) if torch.cuda.is_available(): pipe = pipe.to("cuda") def inference(prompt, guidance, steps): prompt = prompt_prefixes[current_model] + prompt image = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512).images[0] return image with gr.Blocks() as demo: gr.HTML( """
""" ) with gr.Row(): with gr.Column(): model = gr.Dropdown(label="Model", choices=models, value=models[0]) prompt = gr.Textbox(label="Prompt", placeholder="{} is added automatically".format(prompt_prefixes[current_model])) guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15) steps = gr.Slider(label="Steps", value=50, maximum=100, minimum=2) run = gr.Button(value="Run") gr.Markdown(f"Running on: {device}") with gr.Column(): image_out = gr.Image(height=512) model.change(on_model_change, inputs=model, outputs=[]) run.click(inference, inputs=[prompt, guidance, steps], outputs=image_out) gr.Examples([ ["portrait of a girl in dgs illustration style, Anime girl, female soldier working in a cyberpunk city, cleavage, ((perfect femine face)), intricate, 8k, highly detailed, shy, digital painting, intense, sharp focus", 7, 20], ["a photo of muscular beard soldier male in dgs illustration style, half-body, holding robot arms, strong chest", 7.0, 20], ["portrait of ((Harry Potter)) muscular ((male)) in dgs illustration style, photorealistic painting, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20], ["portrait of (liu yifei) girl in dgs illustration style, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20], ["portrait of in dgs illustration style, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20], ], [prompt, guidance, steps], image_out, inference, cache_examples=torch.cuda.is_available()) gr.HTML('''Model by @dgspitzer ❤️