Dianor commited on
Commit
a4546ce
·
verified ·
1 Parent(s): 9b4b972

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -16,12 +16,12 @@ logging.basicConfig(level=logging.INFO)
16
  logger = logging.getLogger(__name__)
17
 
18
  # Проверка наличия токена
19
- if "HUGGINGFACE_TOKEN" not in os.environ:
20
- logger.error("HUGGINGFACE_TOKEN not found in environment variables!")
21
  else:
22
- logger.info("HUGGINGFACE_TOKEN found")
23
  # Аутентификация с использованием токена
24
- login(token=os.environ["HUGGINGFACE_TOKEN"])
25
  logger.info("Logged in to Hugging Face")
26
 
27
  # Определение размера изображения
@@ -38,14 +38,14 @@ class SecureModel:
38
  model_path = hf_hub_download(
39
  repo_id="Dianor/trading-model-private",
40
  filename="trading_modelbeta0.7.keras",
41
- token=os.environ["HUGGINGFACE_TOKEN"]
42
  )
43
 
44
  # Загружаем файл с кастомными слоями
45
  layers_path = hf_hub_download(
46
  repo_id="Dianor/trading-model-private",
47
  filename="custom_trading_layers.py",
48
- token=os.environ["HUGGINGFACE_TOKEN"]
49
  )
50
 
51
  logger.info(f"Files downloaded successfully")
 
16
  logger = logging.getLogger(__name__)
17
 
18
  # Проверка наличия токена
19
+ if "HUGGINGFACE_TOKEN2" not in os.environ:
20
+ logger.error("HUGGINGFACE_TOKEN2 not found in environment variables!")
21
  else:
22
+ logger.info("HUGGINGFACE_TOKEN2 found")
23
  # Аутентификация с использованием токена
24
+ login(token=os.environ["HUGGINGFACE_TOKEN2"])
25
  logger.info("Logged in to Hugging Face")
26
 
27
  # Определение размера изображения
 
38
  model_path = hf_hub_download(
39
  repo_id="Dianor/trading-model-private",
40
  filename="trading_modelbeta0.7.keras",
41
+ token=os.environ["HUGGINGFACE_TOKEN2"]
42
  )
43
 
44
  # Загружаем файл с кастомными слоями
45
  layers_path = hf_hub_download(
46
  repo_id="Dianor/trading-model-private",
47
  filename="custom_trading_layers.py",
48
+ token=os.environ["HUGGINGFACE_TOKEN2"]
49
  )
50
 
51
  logger.info(f"Files downloaded successfully")