Art_Generator / app.py
Keiraj's picture
Update app.py
f94bc9a verified
raw
history blame
301 Bytes
import subprocess
subprocess.run(["pip", "install", "diffusers"])
from diffusers import DiffusionPipeline
pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
prompt = input("What do you want to see?")
image = pipe(prompt).images[0]
image
print("How cool is this!")