Fix Search

backend/semantic_search.py

@@ -3,6 +3,7 @@ import lancedb
 import os
 from pathlib import Path
 from sentence_transformers import SentenceTransformer
+from transformers import AutoModel
 
 # EMB_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
 EMB_MODEL_NAME = "jinaai/jina-embeddings-v2-base-en"
@@ -11,6 +12,7 @@ DB_TABLE_NAME = "chunks"
 # Setting up the logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
+model = AutoModel.from_pretrained(EMB_MODEL_NAME, trust_remote_code=True)
 retriever = SentenceTransformer(EMB_MODEL_NAME)
 
 # db