helblazer811 commited on
Commit
c02178b
·
1 Parent(s): c8c672f

Added access token read from command line

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -9,6 +9,14 @@ import os
9
 
10
  from concept_attention import ConceptAttentionFluxPipeline
11
 
 
 
 
 
 
 
 
 
12
  IMG_SIZE = 210
13
  COLUMNS = 5
14
 
 
9
 
10
  from concept_attention import ConceptAttentionFluxPipeline
11
 
12
+ import os
13
+ from huggingface_hub import login
14
+
15
+ # Load token from environment
16
+ hf_token = os.getenv("AccessToken")
17
+ # Log in with the token (this sets it globally in the session)
18
+ login(token=hf_token)
19
+
20
  IMG_SIZE = 210
21
  COLUMNS = 5
22