anl139 commited on
Commit
1f27a13
·
verified ·
1 Parent(s): afb88fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -13
app.py CHANGED
@@ -94,19 +94,12 @@ for doc in data:
94
 
95
  # Create a Chroma vector store using the document splits.
96
  persist_directory = "./chroma_db"
97
- if os.path.exists(persist_directory) and os.listdir(persist_directory):
98
- vectorstore = Chroma(
99
- persist_directory=persist_directory,
100
- embedding_function=OpenAIEmbeddings()
101
- )
102
- print("Loaded vector store from persist directory.")
103
- else:
104
- vectorstore = Chroma.from_documents(
105
- documents=all_splits,
106
- embedding=OpenAIEmbeddings(),
107
- persist_directory=persist_directory
108
- )
109
- print("Created new vector store and persisted embeddings.")
110
 
111
  # Create a BM25 retriever from the document splits.
112
  bm25_retriever = BM25Retriever.from_documents(all_splits, search_kwargs={"k": 5})
 
94
 
95
  # Create a Chroma vector store using the document splits.
96
  persist_directory = "./chroma_db"
97
+ vectorstore = Chroma.from_documents(
98
+ documents=all_splits,
99
+ embedding=OpenAIEmbeddings(),
100
+ persist_directory=persist_directory
101
+ )
102
+ print("Created new vector store and persisted embeddings.")
 
 
 
 
 
 
 
103
 
104
  # Create a BM25 retriever from the document splits.
105
  bm25_retriever = BM25Retriever.from_documents(all_splits, search_kwargs={"k": 5})