pavvloff commited on
Commit
af26a9d
1 Parent(s): 9e7a255

Update backend/semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +2 -2
backend/semantic_search.py CHANGED
@@ -45,9 +45,9 @@ def openai_embedding(text, key = None):
45
  return rs.data[0].embedding
46
 
47
  minilm = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
48
- gtelarge = SentenceTransformer('thenlper/gte-large')
49
  retrievers['MiniLM'] = lambda t, key: minilm.encode(t)
50
- retrievers['GteLarge'] = lambda t, key: gtelarge.encode(t)
51
  retrievers['OpenAI'] = openai_embedding
52
 
53
  # db
 
45
  return rs.data[0].embedding
46
 
47
  minilm = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
48
+ mpnet = SentenceTransformer('sentence-transformers/all-mpnet-base-v2')
49
  retrievers['MiniLM'] = lambda t, key: minilm.encode(t)
50
+ retrievers['mpnet'] = lambda t, key: mpnet.encode(t)
51
  retrievers['OpenAI'] = openai_embedding
52
 
53
  # db