VsCode commited on
Commit
12e96ed
Β·
1 Parent(s): 65f4d38

solving conflit in login function again 5

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -33,10 +33,17 @@ pipe = None
33
 
34
  import os
35
  from transformers import pipeline
 
 
 
 
 
 
 
36
  @app.on_event("startup")
37
  async def startup_event():
38
  global pipe
39
- token = os.environ.get("HF_TOKEN")
40
  if not token:
41
  raise RuntimeError("HF_TOKEN nΓ£o encontrado no ambiente")
42
  pipe = pipeline(
 
33
 
34
  import os
35
  from transformers import pipeline
36
+
37
+ def get_hf_token():
38
+ return (
39
+ os.environ.get("HUGGINGFACE_HUB_TOKEN")
40
+ or os.environ.get("HF_TOKEN")
41
+ )
42
+
43
  @app.on_event("startup")
44
  async def startup_event():
45
  global pipe
46
+ token = get_hf_token()
47
  if not token:
48
  raise RuntimeError("HF_TOKEN nΓ£o encontrado no ambiente")
49
  pipe = pipeline(