christinac commited on
Commit
506ff08
1 Parent(s): 20943b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ from diffusers import StableDiffusionImg2ImgPipeline
11
 
12
  YOUR_TOKEN=MY_SECRET_TOKEN
13
  device="cpu"
14
- img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=YOUR_TOKEN)
15
- img_pipe.to(device)
16
 
17
  #draw an image based off of user's text input
18
 
 
11
 
12
  YOUR_TOKEN=MY_SECRET_TOKEN
13
  device="cpu"
14
+ pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=YOUR_TOKEN)
15
+ pipe.to(device)
16
 
17
  #draw an image based off of user's text input
18