Art_Generator / app.py
Keiraj's picture
Update app.py
162eb52 verified
raw
history blame
404 Bytes
import subprocess
subprocess.run(["pip", "install", "diffusers"])
subprocess.check_call([sys.executable, "-m", "pip", "install", "torch", "torchvision", "torchaudio"])
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!")