Sylvain Filoni commited on
Commit
314e663
โ€ข
1 Parent(s): 36960e6
Files changed (2) hide show
  1. constants.py +1 -1
  2. utils.py +2 -2
constants.py CHANGED
@@ -1,7 +1,7 @@
1
  import numpy as np
2
  import os
3
 
4
- MUBERT_LICENCE = os.environ.get('MUBERT_LICENCE')
5
  MUBERT_TOKEN = os.environ.get('MUBERT_TOKEN')
6
 
7
  MUBERT_MODE = "loop"
1
  import numpy as np
2
  import os
3
 
4
+ MUBERT_LICENSE = os.environ.get('MUBERT_LICENSE')
5
  MUBERT_TOKEN = os.environ.get('MUBERT_TOKEN')
6
 
7
  MUBERT_MODE = "loop"
utils.py CHANGED
@@ -3,7 +3,7 @@ import numpy as np
3
  import httpx
4
  import os
5
 
6
- from constants import MUBERT_TAGS, MUBERT_MODE, MUBERT_LICENCE, MUBERT_TOKEN
7
 
8
  def get_mubert_tags_embeddings(w2v_model):
9
  return w2v_model.encode(MUBERT_TAGS)
@@ -15,7 +15,7 @@ def get_pat(email: str):
15
  "method": "GetServiceAccess",
16
  "params": {
17
  "email": email,
18
- "license": MUBERT_LICENCE,
19
  "token": MUBERT_TOKEN,
20
  "mode": MUBERT_MODE,
21
  }
3
  import httpx
4
  import os
5
 
6
+ from constants import MUBERT_TAGS, MUBERT_MODE, MUBERT_LICENSE, MUBERT_TOKEN
7
 
8
  def get_mubert_tags_embeddings(w2v_model):
9
  return w2v_model.encode(MUBERT_TAGS)
15
  "method": "GetServiceAccess",
16
  "params": {
17
  "email": email,
18
+ "license": MUBERT_LICENSE,
19
  "token": MUBERT_TOKEN,
20
  "mode": MUBERT_MODE,
21
  }