rameshmoorthy commited on
Commit
ed59c02
1 Parent(s): e29dcdc

Update backend/semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +2 -2
backend/semantic_search.py CHANGED
@@ -15,14 +15,14 @@ task = "qa" # Encode for a specific task (qa, icl, chat, lrlm, tool, convsearch)
15
 
16
  #EMB_MODEL_NAME = "thenlper/gte-base"
17
  #DB_TABLE_NAME = "Huggingface_docs"
18
- DB_TABLE_NAME = "my_table"
19
  # Setting up the logging
20
  logging.basicConfig(level=logging.INFO)
21
  logger = logging.getLogger(__name__)
22
  retriever = SentenceTransformer(EMB_MODEL_NAME)
23
 
24
  # db
25
- db_uri = os.path.join(Path(__file__).parents[1], ".lancedb1")
26
  print(f'DB URL is {db_uri}')
27
  db = lancedb.connect(db_uri)
28
  table = db.open_table(DB_TABLE_NAME)
 
15
 
16
  #EMB_MODEL_NAME = "thenlper/gte-base"
17
  #DB_TABLE_NAME = "Huggingface_docs"
18
+ DB_TABLE_NAME = "cbse"
19
  # Setting up the logging
20
  logging.basicConfig(level=logging.INFO)
21
  logger = logging.getLogger(__name__)
22
  retriever = SentenceTransformer(EMB_MODEL_NAME)
23
 
24
  # db
25
+ db_uri = os.path.join(Path(__file__).parents[1], ".lancedb")
26
  print(f'DB URL is {db_uri}')
27
  db = lancedb.connect(db_uri)
28
  table = db.open_table(DB_TABLE_NAME)