lauer commited on
Commit
3d14c7e
1 Parent(s): 94a8f6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ import requests
14
  from share_btn import community_icon_html, loading_icon_html, share_js
15
 
16
  model_id = "CompVis/stable-diffusion-v1-4"
17
- device = "CPU"
18
- auth_token = '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'
19
 
20
  #If you are running this code locally, you need to either do a 'huggingface-cli login` or paste your User Access Token from here https://huggingface.co/settings/tokens into the use_auth_token field below.
21
  pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token="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", revision="fp16", torch_dtype=torch.float16)
14
  from share_btn import community_icon_html, loading_icon_html, share_js
15
 
16
  model_id = "CompVis/stable-diffusion-v1-4"
17
+ device = "cpu"
18
+ auth_token = os.getenv("auth_token")
19
 
20
  #If you are running this code locally, you need to either do a 'huggingface-cli login` or paste your User Access Token from here https://huggingface.co/settings/tokens into the use_auth_token field below.
21
  pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token="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", revision="fp16", torch_dtype=torch.float16)