Spaces:
Sleeping
Sleeping
Update vectorstore.py
Browse files- 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(
|
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)
|