Upload app.py
Browse files
app.py
CHANGED
@@ -90,7 +90,7 @@ def predict(user_input,company):
|
|
90 |
}
|
91 |
|
92 |
filter = "dataset/"+company+"-10-k-2023.pdf"
|
93 |
-
|
94 |
|
95 |
# Create context_for_query
|
96 |
relevant_document_chunks = retriever.get_relevant_documents(user_question)
|
|
|
90 |
}
|
91 |
|
92 |
filter = "dataset/"+company+"-10-k-2023.pdf"
|
93 |
+
retreiver = vectorstore_persisted.similarity_search(user_input, k=5, filter={"source":filter})
|
94 |
|
95 |
# Create context_for_query
|
96 |
relevant_document_chunks = retriever.get_relevant_documents(user_question)
|