Rulga commited on
Commit
4f531f0
·
1 Parent(s): 92f4eab

Refactor settings.py: Update token retrieval to use 'pro_api_token' and improve error message for missing token

Browse files
Files changed (1) hide show
  1. config/settings.py +5 -3
config/settings.py CHANGED
@@ -3,9 +3,11 @@ import tempfile
3
  from huggingface_hub import HfApi
4
 
5
  # API tokens
6
- HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 
7
  if not HF_TOKEN:
8
- raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
 
9
 
10
  # API Configuration
11
  API_CONFIG = {
@@ -16,7 +18,7 @@ API_CONFIG = {
16
  "headers": {
17
  "X-Use-Cache": "false",
18
  "Content-Type": "application/json",
19
- "Authorization": f"Bearer {HF_TOKEN}" # явно указываем авторизацию
20
  }
21
  }
22
 
 
3
  from huggingface_hub import HfApi
4
 
5
  # API tokens
6
+ #HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
7
+ HF_TOKEN = os.getenv("pro_api_token")
8
  if not HF_TOKEN:
9
+ raise ValueError("pro_api_token not found in environment variables")
10
+ #raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
11
 
12
  # API Configuration
13
  API_CONFIG = {
 
18
  "headers": {
19
  "X-Use-Cache": "false",
20
  "Content-Type": "application/json",
21
+ "Authorization": f"Bearer {HF_TOKEN}"
22
  }
23
  }
24