red1xe commited on
Commit
035f699
·
1 Parent(s): 234ded3

chrome problem fixed

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ def get_pdf_text_chunks(pdf_text):
24
 
25
  def create_vector_store(target_source_chunks):
26
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model_name)
27
- db = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
28
- db.add(target_source_chunks)
29
  return db
30
 
31
  def get_vector_store(target_source_chunks):
 
24
 
25
  def create_vector_store(target_source_chunks):
26
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model_name)
27
+ db = Chroma.from_documents(documents=target_source_chunks, persist_directory=persist_directory, embedding=embeddings)
28
+ db.persist()
29
  return db
30
 
31
  def get_vector_store(target_source_chunks):