Update app.py
Browse files
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 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 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})
|