superbsaeed commited on
Commit
5c08fcb
·
verified ·
1 Parent(s): 5099d05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import FluxPipeline
5
  import os
6
 
7
  # 1. Initialize the Pipeline
8
- hf_token = os.getenv("HF_TOKEN")
9
  model_id = "black-forest-labs/FLUX.1-schnell"
10
 
11
  # Load the model with bfloat16 to save memory and improve speed
 
5
  import os
6
 
7
  # 1. Initialize the Pipeline
8
+ hf_token = os.getenv("FLUX_API_TOKEN")
9
  model_id = "black-forest-labs/FLUX.1-schnell"
10
 
11
  # Load the model with bfloat16 to save memory and improve speed