kayfahaarukku commited on
Commit
910d80a
1 Parent(s): 5b6ebc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,6 +5,8 @@ import gradio as gr
5
  import random
6
  import tqdm
7
 
 
 
8
  # Enable TQDM progress tracking
9
  tqdm.monitor_interval = 0
10
 
@@ -14,7 +16,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
14
  torch_dtype=torch.float16,
15
  custom_pipeline="lpw_stable_diffusion_xl",
16
  use_safetensors=True,
17
- use_auth=HUGGING_FACE_HUB_TOKEN,
18
  )
19
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
20
 
 
5
  import random
6
  import tqdm
7
 
8
+ HF_TOKEN = os.getenv("HF_TOKEN")
9
+
10
  # Enable TQDM progress tracking
11
  tqdm.monitor_interval = 0
12
 
 
16
  torch_dtype=torch.float16,
17
  custom_pipeline="lpw_stable_diffusion_xl",
18
  use_safetensors=True,
19
+ use_auth=HF_TOKEN,
20
  )
21
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
22