Spaces:
Running
on
Zero
Running
on
Zero
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()]
|