bespin-global
/

bb8-embedder

Sentence Similarity

sentence-transformers

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

Copycats commited on Jul 3

Commit

f6ef4f5

•

1 Parent(s): a4db2b5

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -7,8 +7,8 @@ device = 0 if torch.cuda.is_available() else -1
 # multi-model list
 multi_model_list = [
-    {"model_id": "BAAI/bge-base-en-v1.5", "task":"sentence-embeddings"},
-    {"model_id": "BAAI/bge-reranker-base", "task":"sentence-ranking"},
 ]
 class EndpointHandler():

 # multi-model list
 multi_model_list = [
+    {"model_id": "BAAI/bge-base-en-v1.5", "task":"feature-extraction"},
+    {"model_id": "BAAI/bge-reranker-base", "task":"text-classification"},
 ]
 class EndpointHandler():