Update app.py

app.py

@@ -149,7 +149,8 @@ def predict(user_input,company):
 
     # Create context_for_query
 
-    relevant_document_chunks = retriever.invoke(user_input)
+    filter = "dataset/"+company+"-10-k-2023.pdf"
+    relevant_document_chunks = retriever.invoke(user_input, k=5, filter={"source":filter})
     
     context_list = [d.page_content for d in relevant_document_chunks]
     context_for_query = ". ".join(context_list)