Shuang59 commited on
Commit
5a31388
β€’
1 Parent(s): 189b1c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,6 +7,9 @@ Original file is located at
7
  https://colab.research.google.com/drive/19xx6Nu4FeiGj-TzTUFxBf-15IkeuFx_F
8
  """
9
 
 
 
 
10
  import gradio as gr
11
  import torch as th
12
 
@@ -28,7 +31,7 @@ print(device)
28
  # init stable diffusion model
29
  pipe = StableDiffusionPipeline.from_pretrained(
30
  "CompVis/stable-diffusion-v1-4",
31
- use_auth_token='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'
32
  ).to(device)
33
 
34
 
 
7
  https://colab.research.google.com/drive/19xx6Nu4FeiGj-TzTUFxBf-15IkeuFx_F
8
  """
9
 
10
+ import os
11
+ USER_TOKEN = os.environ.getattribute("USER_TOKEN")
12
+
13
  import gradio as gr
14
  import torch as th
15
 
 
31
  # init stable diffusion model
32
  pipe = StableDiffusionPipeline.from_pretrained(
33
  "CompVis/stable-diffusion-v1-4",
34
+ use_auth_token=USER_TOKEN
35
  ).to(device)
36
 
37