Hansimov commited on
Commit
0ebbe50
1 Parent(s): 48cf048

:boom: [Fix] Missing HF_ENDPOINT config in hf space

Browse files
Files changed (2) hide show
  1. configs/info.json +1 -1
  2. transforms/embed.py +2 -1
configs/info.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "version": "0.1.2",
3
  "app_name": "Embedding API",
4
  "server": "0.0.0.0",
5
  "port": 16666
 
1
  {
2
+ "version": "0.1.3",
3
  "app_name": "Embedding API",
4
  "server": "0.0.0.0",
5
  "port": 16666
transforms/embed.py CHANGED
@@ -9,7 +9,8 @@ from numpy.linalg import norm
9
  from configs.envs import ENVS
10
  from configs.constants import AVAILABLE_MODELS
11
 
12
- os.environ["HF_ENDPOINT"] = ENVS["HF_ENDPOINT"]
 
13
  os.environ["HF_TOKEN"] = ENVS["HF_TOKEN"]
14
 
15
 
 
9
  from configs.envs import ENVS
10
  from configs.constants import AVAILABLE_MODELS
11
 
12
+ if ENVS["HF_ENDPOINT"]:
13
+ os.environ["HF_ENDPOINT"] = ENVS["HF_ENDPOINT"]
14
  os.environ["HF_TOKEN"] = ENVS["HF_TOKEN"]
15
 
16