bupa1018 commited on
Commit
ae6cea7
·
1 Parent(s): 2227dd0

Update vectorstore.py

Browse files
Files changed (1) hide show
  1. vectorstore.py +1 -1
vectorstore.py CHANGED
@@ -1,7 +1,7 @@
1
  from langchain_community.vectorstores import Chroma
2
 
3
  # Setup Vectorstore
4
- def setup_vectorstore(chunks, model_name, persist_directory):
5
  print("Start setup_vectorstore_function")
6
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
7
  vectorstore = get_chroma_vectorstore(embedding_model, persist_directory)
 
1
  from langchain_community.vectorstores import Chroma
2
 
3
  # Setup Vectorstore
4
+ def setup_vectorstore(docs, model_name, persist_directory):
5
  print("Start setup_vectorstore_function")
6
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
7
  vectorstore = get_chroma_vectorstore(embedding_model, persist_directory)