Spaces:
Build error
Build error
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 |
|