Hugging Face's logo

Spaces:
ThinkAI-Morocco
/
TESTing
Sleeping

App Files Community
AkramOM606 commited on
Commit
9eae405
·
verified ·
1 Parent(s): bd6e1da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import transformers
2
  import torch
3
  import os
4
 
5
- os.environ["HF_TOKEN"] = window.huggingface.variables.token
6
- os.environ["HUGGINGFACEHUB_API_TOKEN"] = window.huggingface.variables.token
7
  # os.environ["USE_FLASH_ATTENTION"] = "1"
8
 
9
  print(f"Device name: {torch.cuda.get_device_properties('cuda').name}")
 
2
  import torch
3
  import os
4
 
5
+ os.environ["HF_TOKEN"] = st.secrets["HF_TOKEN"]
6
+ os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets["HF_TOKEN"]
7
  # os.environ["USE_FLASH_ATTENTION"] = "1"
8
 
9
  print(f"Device name: {torch.cuda.get_device_properties('cuda').name}")