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!")