Keiraj commited on
Commit
162eb52
·
verified ·
1 Parent(s): f94bc9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import subprocess
2
  subprocess.run(["pip", "install", "diffusers"])
3
 
 
 
4
  from diffusers import DiffusionPipeline
5
  pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
6
  prompt = input("What do you want to see?")
 
1
  import subprocess
2
  subprocess.run(["pip", "install", "diffusers"])
3
 
4
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "torch", "torchvision", "torchaudio"])
5
+
6
  from diffusers import DiffusionPipeline
7
  pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
8
  prompt = input("What do you want to see?")