NCTCMumbai
commited on
Commit
•
eebaf09
1
Parent(s):
4892a2c
Update backend/semantic_search.py
Browse files
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], ".
|
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)
|