Nicolai Berk commited on
Commit
253a65f
·
1 Parent(s): 0ebed34

Add print for FAISS debug

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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()]