bupa1018 commited on
Commit
5f6c8c6
·
1 Parent(s): 85034d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -287,6 +287,10 @@ def setup_vectorstore(chunks, model_name, persist_directory):
287
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
288
 
289
  vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
 
 
 
 
290
  print("test1", vectorstore._persist_directory)
291
  print("test2",vectorstore.__dir__)
292
  return vectorstore
 
287
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
288
 
289
  vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
290
+ path = persist_directory
291
+
292
+ root_dir = os.path.splitdrive(path)[0] or os.sep
293
+ print("Root directory:", root_dir)
294
  print("test1", vectorstore._persist_directory)
295
  print("test2",vectorstore.__dir__)
296
  return vectorstore