Commit
•
4400012
1
Parent(s):
2765734
Update backend/semantic_search.py
Browse files
backend/semantic_search.py
CHANGED
@@ -4,7 +4,7 @@ import lancedb
|
|
4 |
import os
|
5 |
from pathlib import Path
|
6 |
from sentence_transformers import SentenceTransformer
|
7 |
-
from FlagEmbedding import LLMEmbedder, FlagReranker # Al document present here https://github.com/FlagOpen/FlagEmbedding/tree/master
|
8 |
#EMB_MODEL_NAME = "thenlper/gte-base"
|
9 |
EMB_MODEL_NAME = 'BAAI/llm-embedder'
|
10 |
task = "qa" # Encode for a specific task (qa, icl, chat, lrlm, tool, convsearch)
|
|
|
4 |
import os
|
5 |
from pathlib import Path
|
6 |
from sentence_transformers import SentenceTransformer
|
7 |
+
#from FlagEmbedding import LLMEmbedder, FlagReranker # Al document present here https://github.com/FlagOpen/FlagEmbedding/tree/master
|
8 |
#EMB_MODEL_NAME = "thenlper/gte-base"
|
9 |
EMB_MODEL_NAME = 'BAAI/llm-embedder'
|
10 |
task = "qa" # Encode for a specific task (qa, icl, chat, lrlm, tool, convsearch)
|