|
import gradio as gr |
|
import spaces |
|
from diffusers import KandinskyPriorPipeline, KandinskyPipeline |
|
from diffusers.utils import load_image |
|
import torch |
|
from PIL import Image |
|
|
|
pipe_prior = KandinskyPriorPipeline.from_pretrained( |
|
"kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16 |
|
) |
|
pipe_prior.to("cuda") |
|
|
|
pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1", torch_dtype=torch.float16) |
|
pipe.to("cuda") |
|
|
|
|
|
@spaces.GPU() |
|
def squarify_image(img): |
|
if(img.height > img.width): bg_size = img.height |
|
else: bg_size = img.width |
|
bg = Image.new(mode="RGB", size=(bg_size,bg_size), color="white") |
|
bg.paste(img, ( int((bg.width - bg.width)/2), 0) ) |
|
|
|
return bg |
|
|
|
@spaces.GPU() |
|
def blend(img1, img2, slider): |
|
img1.thumbnail((1024, 1024)) |
|
img2.thumbnail((1024, 1024)) |
|
|
|
img1 = squarify_image(img1) |
|
img2 = squarify_image(img2) |
|
|
|
|
|
images_texts = [img1, img2] |
|
|
|
|
|
weights = [1-slider, slider] |
|
|
|
prior_out = pipe_prior.interpolate(images_texts, weights) |
|
|
|
image = pipe(prompt='', **prior_out, height=1024, width=1024).images[0] |
|
|
|
return image |
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown(""" |
|
# Image Blender |
|
by [Tony Assi](https://www.tonyassi.com/) |
|
""") |
|
|
|
with gr.Row(): |
|
with gr.Column(): |
|
img1 = gr.Image(label='Image 0', type='pil') |
|
img2 = gr.Image(label='Image 1',type='pil') |
|
slider = gr.Slider(label='Weight', maximum=1.0, value=0.5) |
|
btn = gr.Button("Blend") |
|
with gr.Column(): |
|
output = gr.Image(label='Result') |
|
|
|
gr.Examples( |
|
[['./cat.png', './starry_night.jpg', 0.5]], |
|
[img1, img2, slider], |
|
output, |
|
blend, |
|
cache_examples=True, |
|
) |
|
|
|
btn.click(fn=blend, inputs=[img1, img2, slider], outputs=output) |
|
|
|
demo.launch() |