radu.mutilica
commited on
Commit
·
17c9174
1
Parent(s):
b352fd9
modified config, debugging initialization issues
Browse files- config.py +0 -1
- 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 |
-
|
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(
|