Update app.py

app.py

@@ -63,7 +63,7 @@ def bot(history, cross_encoder):
         gr.Warning('Retrieving using ColBERT')
         RAG= RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")
         RAG_db=RAG.from_index('.ragatouille/colbert/indexes/mockingbird')
-        documents_full=RAG_db.search(query)
+        documents_full=RAG_db.search(query,k=top_k_rank)
         
         documents=[item['content'] for item in documents_full]
         # Create Prompt