| 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() |
