Spaces:
Runtime error
Runtime error
File size: 2,199 Bytes
06e80a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import os
import gradio as gr
from inference import load_models, cache_path
from PIL import Image
from os import path
canvas_size = 512
if not path.exists(cache_path):
os.makedirs(cache_path, exist_ok=True)
with gr.Blocks() as demo:
infer = load_models()
with gr.Column():
with gr.Row():
with gr.Column():
s = gr.Slider(label="steps", minimum=4, maximum=8, step=1, value=4, interactive=True)
c = gr.Slider(label="cfg", minimum=0.1, maximum=3, step=0.1, value=1, interactive=True)
i_s = gr.Slider(label="sketch strength", minimum=0.1, maximum=0.9, step=0.1, value=0.9, interactive=True)
with gr.Column():
mod = gr.Text(label="Model HuggingFace id (after changing this wait until the model downloads in the console)", value="Lykon/dreamshaper-7", interactive=True)
t = gr.Text(label="Prompt", value="Scary warewolf, 8K, realistic, colorful, long sharp teeth, splash art", interactive=True)
se = gr.Number(label="seed", value=1337, interactive=True)
with gr.Row(equal_height=True):
i = gr.Image(source="canvas", tool="color-sketch", shape=(canvas_size, canvas_size), width=canvas_size, height=canvas_size, type="pil")
o = gr.Image(width=canvas_size, height=canvas_size)
def process_image(p, im, steps, cfg, image_strength, seed):
if not im:
return Image.new("RGB", (canvas_size, canvas_size))
return infer(
prompt=p,
image=im,
num_inference_steps=steps,
guidance_scale=cfg,
strength=image_strength,
seed=int(seed)
)
reactive_controls = [t, i, s, c, i_s, se]
for control in reactive_controls:
control.change(fn=process_image, inputs=reactive_controls, outputs=o)
def update_model(model_name):
global infer
infer = load_models(model_name)
mod.change(fn=update_model, inputs=mod)
if __name__ == "__main__":
demo.launch()
|