Spaces:
Paused
Paused
Update src/retriever.py
Browse files- 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})
|