Spaces:

DaniilAlpha
/

answerer-api

Paused

DaniilAlpha commited on Nov 26, 2023

Commit

d9199ae

1 Parent(s): 3033de7

Update mapper.py

Files changed (1) hide show

mapper.py CHANGED Viewed

@@ -2,7 +2,7 @@ from sentence_transformers import SentenceTransformer, util
 class Mapper:
   def __init__(self, model: str):
-    self.__model = SentenceTransformer(f"models/{model}/pytorch_model.bin")
   def __call__(self, query: str, data: list[str]):
     query_emb = self.__model.encode(query)

 class Mapper:
   def __init__(self, model: str):
+    self.__model = SentenceTransformer(f"models/{model}/")
   def __call__(self, query: str, data: list[str]):
     query_emb = self.__model.encode(query)