Zwicky18's picture
Update app.py
aa91cc8
raw
history blame contribute delete
332 Bytes
from diffusers import StableDiffusionPipeline
import torch
from diffusers import DiffusionPipeline
model_id = "runwayml/stable-diffusion-v1-5"
pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
pipe = pipe.to("cuda")
prompt = "stabilo waifu"
image = pipe(prompt).images[0]
image.save("./stabilo_waifu.png")