Dianor commited on
Commit
b391910
·
verified ·
1 Parent(s): c0a68f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -14
app.py CHANGED
@@ -6,7 +6,6 @@ import cv2
6
  from PIL import Image
7
  import logging
8
  from huggingface_hub import hf_hub_download
9
- from huggingface_hub import login
10
  import matplotlib.pyplot as plt
11
  import matplotlib
12
  matplotlib.use('Agg')
@@ -15,15 +14,6 @@ matplotlib.use('Agg')
15
  logging.basicConfig(level=logging.INFO)
16
  logger = logging.getLogger(__name__)
17
 
18
- # Проверка наличия токена
19
- if "HUGGINGFACE_TOKEN3" not in os.environ:
20
- logger.error("HUGGINGFACE_TOKEN3 not found in environment variables!")
21
- else:
22
- logger.info("HUGGINGFACE_TOKEN3 found")
23
- # Аутентификация с использованием токена
24
- login(token=os.environ["HUGGINGFACE_TOKEN3"])
25
- logger.info("Logged in to Hugging Face")
26
-
27
  # Определение размера изображения
28
  IMG_SHAPE = (479, 1221, 3)
29
 
@@ -37,15 +27,13 @@ class SecureModel:
37
  # Загружаем файл модели
38
  model_path = hf_hub_download(
39
  repo_id="Dianor/trading-model-private",
40
- filename="trading_modelbeta0.7.keras",
41
- token=os.environ["HUGGINGFACE_TOKEN3"]
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_TOKEN3"]
49
  )
50
 
51
  logger.info(f"Files downloaded successfully")
 
6
  from PIL import Image
7
  import logging
8
  from huggingface_hub import hf_hub_download
 
9
  import matplotlib.pyplot as plt
10
  import matplotlib
11
  matplotlib.use('Agg')
 
14
  logging.basicConfig(level=logging.INFO)
15
  logger = logging.getLogger(__name__)
16
 
 
 
 
 
 
 
 
 
 
17
  # Определение размера изображения
18
  IMG_SHAPE = (479, 1221, 3)
19
 
 
27
  # Загружаем файл модели
28
  model_path = hf_hub_download(
29
  repo_id="Dianor/trading-model-private",
30
+ filename="trading_modelbeta0.7.keras"
 
31
  )
32
 
33
  # Загружаем файл с кастомными слоями
34
  layers_path = hf_hub_download(
35
  repo_id="Dianor/trading-model-private",
36
+ filename="custom_trading_layers.py"
 
37
  )
38
 
39
  logger.info(f"Files downloaded successfully")