Hugging Face's logo

Spaces:
JustusI
/
data_roles
Sleeping

App Files Community
JustusI commited on
Commit
b24aade
·
verified ·
1 Parent(s): 307744d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def create_vector_db(docs):
34
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
35
  texts = text_splitter.split_documents(docs)
36
  embedding_function = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
37
- vectordb = Chroma.from_documents(docs, embedding_function)
38
  vectordb.persist()
39
  vectordb = None
40
  vectordb = Chroma(persist_directory=vectordb, embedding_function=embedding_function)
 
34
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
35
  texts = text_splitter.split_documents(docs)
36
  embedding_function = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
37
+ vectordb = Chroma.from_documents(docs, embedding_function,persist_directory='./chroma_db')
38
  vectordb.persist()
39
  vectordb = None
40
  vectordb = Chroma(persist_directory=vectordb, embedding_function=embedding_function)