Yadukrishnan commited on
Commit
84fee95
·
verified ·
1 Parent(s): 6a5b6b1

Update src/retriever.py

Browse files
Files changed (1) hide show
  1. src/retriever.py +1 -1
src/retriever.py CHANGED
@@ -10,7 +10,7 @@ cached_retriever = None
10
  def setup_retriever():
11
  global cached_retriever
12
  if cached_retriever is None:
13
- db = FAISS.load_local(DB_PATH, HuggingFaceEmbeddings(model_name='BAAI/bge-base-en-v1.5'))
14
  with open(METADATA_PATH, "rb") as f:
15
  chunked_docs = pickle.load(f)
16
  cached_retriever = db.as_retriever(search_type="similarity", search_kwargs={'k': 4})
 
10
  def setup_retriever():
11
  global cached_retriever
12
  if cached_retriever is None:
13
+ db = FAISS.load_local(DB_PATH, HuggingFaceEmbeddings(model_name='BAAI/bge-base-en-v1.5'), allow_dangerous_deserialization=True)
14
  with open(METADATA_PATH, "rb") as f:
15
  chunked_docs = pickle.load(f)
16
  cached_retriever = db.as_retriever(search_type="similarity", search_kwargs={'k': 4})