Update app.py
Browse files
app.py
CHANGED
@@ -78,7 +78,7 @@ def main():
|
|
78 |
st.header("Chat with multiple PDFs :books:")
|
79 |
|
80 |
|
81 |
-
|
82 |
user_question = st.text_input("Ask a question about your documents:")
|
83 |
if "messages" not in st.session_state:
|
84 |
st.session_state["messages"] = [
|
@@ -89,7 +89,7 @@ def main():
|
|
89 |
st.chat_message(msg["role"]).write(msg["content"])
|
90 |
|
91 |
|
92 |
-
|
93 |
|
94 |
if user_question:
|
95 |
handle_userinput(user_question,retriever)
|
|
|
78 |
st.header("Chat with multiple PDFs :books:")
|
79 |
|
80 |
|
81 |
+
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20)
|
82 |
user_question = st.text_input("Ask a question about your documents:")
|
83 |
if "messages" not in st.session_state:
|
84 |
st.session_state["messages"] = [
|
|
|
89 |
st.chat_message(msg["role"]).write(msg["content"])
|
90 |
|
91 |
|
92 |
+
|
93 |
|
94 |
if user_question:
|
95 |
handle_userinput(user_question,retriever)
|