Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type=
|
|
52 |
st.write("Vector store doesnt exist and will be created now")
|
53 |
loader = DirectoryLoader('./data/', glob="./*.txt", loader_cls=TextLoader)
|
54 |
docs = loader.load()
|
55 |
-
|
56 |
|
57 |
text_splitter = RecursiveCharacterTextSplitter.from_tiktoken_encoder(
|
58 |
chunk_size=chunk_size, chunk_overlap=chunk_overlap,
|
@@ -65,7 +65,7 @@ def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type=
|
|
65 |
vectorstore = Chroma.from_documents(
|
66 |
documents=split_docs, embedding=embeddings, persist_directory=vectorstore_path
|
67 |
)
|
68 |
-
|
69 |
|
70 |
retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k})
|
71 |
|
@@ -91,7 +91,7 @@ def main():
|
|
91 |
st.header("Chat with multiple Lithuanian Law Documents:" ":books:")
|
92 |
|
93 |
st.markdown("Hi, I am Birute (Powered by qwen2-0_5b model), chat assistant, based on republic of Lithuania law documents. You can choose below information retrieval type and how many documents you want to be retrieved.")
|
94 |
-
st.markdown("Available Documents: LR_Civil_Code_2022, LR_Constitution_2022, LR_Criminal_Code_2018, LR_Criminal_Procedure_code_2022,LR_Labour_code_2010. P.S it's a shame that there are no newest documents translations into English... ")
|
95 |
|
96 |
if "messages" not in st.session_state:
|
97 |
st.session_state["messages"] = [
|
|
|
52 |
st.write("Vector store doesnt exist and will be created now")
|
53 |
loader = DirectoryLoader('./data/', glob="./*.txt", loader_cls=TextLoader)
|
54 |
docs = loader.load()
|
55 |
+
|
56 |
|
57 |
text_splitter = RecursiveCharacterTextSplitter.from_tiktoken_encoder(
|
58 |
chunk_size=chunk_size, chunk_overlap=chunk_overlap,
|
|
|
65 |
vectorstore = Chroma.from_documents(
|
66 |
documents=split_docs, embedding=embeddings, persist_directory=vectorstore_path
|
67 |
)
|
68 |
+
|
69 |
|
70 |
retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k})
|
71 |
|
|
|
91 |
st.header("Chat with multiple Lithuanian Law Documents:" ":books:")
|
92 |
|
93 |
st.markdown("Hi, I am Birute (Powered by qwen2-0_5b model), chat assistant, based on republic of Lithuania law documents. You can choose below information retrieval type and how many documents you want to be retrieved.")
|
94 |
+
st.markdown(### "Available Documents: LR_Civil_Code_2022, LR_Constitution_2022, LR_Criminal_Code_2018, LR_Criminal_Procedure_code_2022,LR_Labour_code_2010. P.S it's a shame that there are no newest documents translations into English... ")
|
95 |
|
96 |
if "messages" not in st.session_state:
|
97 |
st.session_state["messages"] = [
|