ravi6k commited on
Commit
c5c21b3
·
verified ·
1 Parent(s): 6e5ff9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def predict(user_input,company):
106
 
107
  # Create context_for_query
108
  user_input = user_input
109
- relevant_document_chunks = vectorstore_retriever.similarity_search(user_input,k=5,filter={"source":filter_company})
110
  context_list = [d.page_content for d in relevant_document_chunks]
111
  context_for_query = ". ".join(context_list)
112
 
 
106
 
107
  # Create context_for_query
108
  user_input = user_input
109
+ relevant_document_chunks = vectorstore_retriever.get_relevant_documents(user_input,k=5,filter={"source":filter_company})
110
  context_list = [d.page_content for d in relevant_document_chunks]
111
  context_for_query = ". ".join(context_list)
112