Spaces:
Sleeping
Sleeping
Upload retrieval.py
Browse files- 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}
|
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)
|