cb1716pics commited on
Commit
861479a
·
verified ·
1 Parent(s): 7bb8fda

Upload retrieval.py

Browse files
Files changed (1) hide show
  1. retrieval.py +1 -1
retrieval.py CHANGED
@@ -12,7 +12,7 @@ def retrieve_documents(query, top_k=5):
12
  query_dataset = find_query_dataset(query)
13
  #index, chunk_docs = load_data_from_faiss(query)
14
 
15
- faiss_index_path = f"data_local/{query_dataset}__quantized.faiss"
16
  index = faiss.read_index(faiss_index_path)
17
 
18
  query_embedding = np.array(embedding_model.embed_documents([query]), dtype=np.float32)
 
12
  query_dataset = find_query_dataset(query)
13
  #index, chunk_docs = load_data_from_faiss(query)
14
 
15
+ faiss_index_path = f"data_local/{query_dataset}_quantized.faiss"
16
  index = faiss.read_index(faiss_index_path)
17
 
18
  query_embedding = np.array(embedding_model.embed_documents([query]), dtype=np.float32)