NCTCMumbai commited on
Commit
eebaf09
1 Parent(s): 4892a2c

Update backend/semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +3 -3
backend/semantic_search.py CHANGED
@@ -6,14 +6,14 @@ from pathlib import Path
6
  from sentence_transformers import SentenceTransformer
7
 
8
  EMB_MODEL_NAME = "thenlper/gte-base"
9
- DB_TABLE_NAME = "Huggingface_docs"
10
-
11
  # Setting up the logging
12
  logging.basicConfig(level=logging.INFO)
13
  logger = logging.getLogger(__name__)
14
  retriever = SentenceTransformer(EMB_MODEL_NAME)
15
 
16
  # db
17
- db_uri = os.path.join(Path(__file__).parents[1], ".lancedb")
18
  db = lancedb.connect(db_uri)
19
  table = db.open_table(DB_TABLE_NAME)
 
6
  from sentence_transformers import SentenceTransformer
7
 
8
  EMB_MODEL_NAME = "thenlper/gte-base"
9
+ #DB_TABLE_NAME = "Huggingface_docs"
10
+ DB_TABLE_NAME = "doc_embed1"
11
  # Setting up the logging
12
  logging.basicConfig(level=logging.INFO)
13
  logger = logging.getLogger(__name__)
14
  retriever = SentenceTransformer(EMB_MODEL_NAME)
15
 
16
  # db
17
+ db_uri = os.path.join(Path(__file__).parents[1], ".lancedb1")
18
  db = lancedb.connect(db_uri)
19
  table = db.open_table(DB_TABLE_NAME)