Spaces:
Runtime error
Runtime error
Update backend/semantic_search.py
Browse files
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 = "
|
| 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], ".
|
| 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)
|