DaniilAlpha commited on
Commit
7155dc1
·
1 Parent(s): 88ac2f4

Update mapper.py

Browse files
Files changed (1) hide show
  1. mapper.py +2 -2
mapper.py CHANGED
@@ -1,8 +1,8 @@
1
  from sentence_transformers import SentenceTransformer, util
2
 
3
  class Mapper:
4
- def __init__(self, repo: str, model: str):
5
- self.__model = SentenceTransformer(f"{repo}/{model}")
6
 
7
  def __call__(self, query: str, data: list[str]):
8
  query_emb = self.__model.encode(query)
 
1
  from sentence_transformers import SentenceTransformer, util
2
 
3
  class Mapper:
4
+ def __init__(self, model: str):
5
+ self.__model = SentenceTransformer(model)
6
 
7
  def __call__(self, query: str, data: list[str]):
8
  query_emb = self.__model.encode(query)