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)
|