Spaces:
Sleeping
Sleeping
File size: 404 Bytes
f94bc9a ac9823f 162eb52 58752cf |
1 2 3 4 5 6 7 8 9 10 11 |
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!") |