bupa1018 commited on
Commit
eff8daf
·
1 Parent(s): 326b887

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
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