NCTCMumbai commited on
Commit
4400012
1 Parent(s): 2765734

Update backend/semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +1 -1
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)