File size: 680 Bytes
798eff9 c2f4706 798eff9 c2f4706 798eff9 c2f4706 798eff9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import gradio as gr
from PIL import Image
import torch
from diffusers import StableDiffusionImg2ImgPipeline
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
"hakurei/waifu-diffusion",
torch_dtype=torch.float32
).to("cpu")
def generate_ghibli_style(image, prompt):
result = pipe(prompt=prompt, image=image, strength=0.6, guidance_scale=7.5)
return result.images[0]
gr.Interface(
fn=generate_ghibli_style,
inputs=[gr.Image(type="pil"), gr.Textbox(label="Prompt")],
outputs=gr.Image(type="pil"),
title="Ghibli Style Generator (Anime Style)",
description="Upload a photo and get a Ghibli-inspired result using waifu-diffusion.",
).launch() |