File size: 553 Bytes
6cab1a9 270bf94 6cab1a9 270bf94 385145e 270bf94 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
import torch
from diffusers import StableDiffusionImg2ImgPipeline
# load the pipeline
device = "cpu"
model_id_or_path = "runwayml/stable-diffusion-v1-5"
# model_id_or_path = "shanetx/stable-diffusion-wtx-app"
img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path)
img2img_pipe = img2img_pipe.to(device)
def img2img_diff(prompt, img):
return img2img_pipe(prompt=prompt, image=img, strength=0.75, guidance_scale=7.5).images[0]
gr.Interface(fn=img2img_diff, inputs=['text', 'image'], outputs='image')
|