Spaces:
Runtime error
Runtime error
Update backend/semantic_search.py
Browse files
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 |
-
|
49 |
retrievers['MiniLM'] = lambda t, key: minilm.encode(t)
|
50 |
-
retrievers['
|
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
|