Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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 |
|