bd

backend/semantic_search.py

@@ -5,8 +5,10 @@ from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import time
+import os
+from pathlib import Path
 
-db = lancedb.connect(".lancedb")
+db = lancedb.connect(Path(__file__).parent / ".lancedb")
 
 TABLE = db.open_table(os.getenv("TABLE_NAME"))
 VECTOR_COLUMN = os.getenv("VECTOR_COLUMN", "vector")