radu.mutilica commited on
Commit
17c9174
·
1 Parent(s): b352fd9

modified config, debugging initialization issues

Browse files
Files changed (2) hide show
  1. config.py +0 -1
  2. handler.py +2 -3
config.py CHANGED
@@ -1,4 +1,3 @@
1
- CROSSENCODER_MODEL = '.'
2
  CROSSENCODER_TOP_K = 6
3
  CROSSENCODER_MAX_LENGTH = 512
4
  CROSSENCODER_MAX_WORKERS = 4
 
 
1
  CROSSENCODER_TOP_K = 6
2
  CROSSENCODER_MAX_LENGTH = 512
3
  CROSSENCODER_MAX_WORKERS = 4
handler.py CHANGED
@@ -31,14 +31,13 @@ class Rank(BaseModel):
31
 
32
 
33
  class EndpointHandler:
34
- def __init__(self, path=""):
35
- crossencoder_model = config.CROSSENCODER_MODEL
36
  self.top_k = config.CROSSENCODER_TOP_K
37
  self.max_length = config.CROSSENCODER_MAX_LENGTH
38
  self.max_workers = config.CROSSENCODER_MAX_WORKERS
39
 
40
  self.model = CrossEncoder(
41
- crossencoder_model,
42
  max_length=self.max_length
43
  )
44
  logger.info(
 
31
 
32
 
33
  class EndpointHandler:
34
+ def __init__(self, path):
 
35
  self.top_k = config.CROSSENCODER_TOP_K
36
  self.max_length = config.CROSSENCODER_MAX_LENGTH
37
  self.max_workers = config.CROSSENCODER_MAX_WORKERS
38
 
39
  self.model = CrossEncoder(
40
+ path,
41
  max_length=self.max_length
42
  )
43
  logger.info(