printing retrieved documents to see what the model is picking

app.py

@@ -66,6 +66,7 @@ def format_prompt(prompt,retrieved_documents,k):
 def talk(prompt,history):
     k = 1 # number of retrieved documents
     scores , retrieved_documents = search(prompt, k)
+    print(retrieved_documents)
     formatted_prompt = format_prompt(prompt,retrieved_documents,k)
     formatted_prompt = formatted_prompt[:2000] # to avoid GPU OOM
     messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}]