Spaces:
Sleeping
Sleeping
Nicolai Berk
commited on
Commit
·
253a65f
1
Parent(s):
0ebed34
Add print for FAISS debug
Browse files
app.py
CHANGED
|
@@ -55,6 +55,11 @@ def rag_pipeline(query):
|
|
| 55 |
# Retrieve top-k from FAISS
|
| 56 |
D, I = index.search(query_embedding, k=5)
|
| 57 |
retrieved_docs = [corpus[idx] for idx in I[0]]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
# Rerank
|
| 60 |
rerank_pairs = [[str(query), str(doc)] for doc in retrieved_docs if isinstance(doc, str) and doc.strip()]
|
|
|
|
| 55 |
# Retrieve top-k from FAISS
|
| 56 |
D, I = index.search(query_embedding, k=5)
|
| 57 |
retrieved_docs = [corpus[idx] for idx in I[0]]
|
| 58 |
+
|
| 59 |
+
print("Retrieved indices:", I[0])
|
| 60 |
+
print("Retrieved docs:")
|
| 61 |
+
for doc in retrieved_docs:
|
| 62 |
+
print("-", repr(doc))
|
| 63 |
|
| 64 |
# Rerank
|
| 65 |
rerank_pairs = [[str(query), str(doc)] for doc in retrieved_docs if isinstance(doc, str) and doc.strip()]
|