Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -7,8 +7,8 @@ device = 0 if torch.cuda.is_available() else -1
|
|
7 |
|
8 |
# multi-model list
|
9 |
multi_model_list = [
|
10 |
-
{"model_id": "BAAI/bge-base-en-v1.5", "task":"
|
11 |
-
{"model_id": "BAAI/bge-reranker-base", "task":"
|
12 |
]
|
13 |
|
14 |
class EndpointHandler():
|
|
|
7 |
|
8 |
# multi-model list
|
9 |
multi_model_list = [
|
10 |
+
{"model_id": "BAAI/bge-base-en-v1.5", "task":"feature-extraction"},
|
11 |
+
{"model_id": "BAAI/bge-reranker-base", "task":"text-classification"},
|
12 |
]
|
13 |
|
14 |
class EndpointHandler():
|