Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -243,15 +243,8 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
|
|
243 |
# Setup Vectorstore
|
244 |
def setup_vectorstore(chunks, model_name, persist_directory):
|
245 |
print("Start setup_vectorstore_function")
|
246 |
-
embedding_model = HuggingFaceEmbeddings(model_name=model_name)
|
247 |
-
|
248 |
vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
|
249 |
-
path = persist_directory
|
250 |
-
|
251 |
-
root_dir = os.path.splitdrive(path)[0] or os.sep
|
252 |
-
print("Root directory:", root_dir)
|
253 |
-
print("test1", vectorstore._persist_directory)
|
254 |
-
print("test2",vectorstore.__dir__)
|
255 |
return vectorstore
|
256 |
|
257 |
|
|
|
243 |
# Setup Vectorstore
|
244 |
def setup_vectorstore(chunks, model_name, persist_directory):
|
245 |
print("Start setup_vectorstore_function")
|
246 |
+
embedding_model = HuggingFaceEmbeddings(model_name=model_name)
|
|
|
247 |
vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
|
|
|
|
|
|
|
|
|
|
|
|
|
248 |
return vectorstore
|
249 |
|
250 |
|