Update services/pdf_service.py
Browse files- services/pdf_service.py +1 -1
services/pdf_service.py
CHANGED
@@ -146,7 +146,7 @@ class PDFService:
|
|
146 |
query_embedding_np = query_embedding.cpu().detach().numpy()
|
147 |
|
148 |
# Search in FAISS index
|
149 |
-
distances, indices = self.
|
150 |
|
151 |
|
152 |
|
|
|
146 |
query_embedding_np = query_embedding.cpu().detach().numpy()
|
147 |
|
148 |
# Search in FAISS index
|
149 |
+
distances, indices = self.index.search(query_embedding_np, top_k)
|
150 |
|
151 |
|
152 |
|