edf commited on
Commit
c8e1db9
1 Parent(s): 51f2141

whoops - embedding_function is for calling Chroma directly, embedding is

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -223,12 +223,12 @@ class TranscriptVectorStore:
223
  def create_or_read_vectorstore(self, persist_dir, rebuild=False):
224
  if persist_dir:
225
  if os.path.isdir(persist_dir) and not rebuild:
226
- self.vs = Chroma(persist_directory=persist_dir, embedding_function=self.embedder)
227
  else:
228
  self.vs = Chroma.from_documents(persist_directory=persist_dir, documents=self.documents, embedding_function=self.embedder)
229
  self.vs.persist()
230
  else:
231
- self.vs = Chroma.from_documents(documents=self.documents, embedding_function=self.embedder)
232
 
233
  # cache list of speakers and times they spoke
234
  self.speakers={}
 
223
  def create_or_read_vectorstore(self, persist_dir, rebuild=False):
224
  if persist_dir:
225
  if os.path.isdir(persist_dir) and not rebuild:
226
+ self.vs = Chroma(persist_directory=persist_dir, embedding=self.embedder)
227
  else:
228
  self.vs = Chroma.from_documents(persist_directory=persist_dir, documents=self.documents, embedding_function=self.embedder)
229
  self.vs.persist()
230
  else:
231
+ self.vs = Chroma.from_documents(documents=self.documents, embedding=self.embedder)
232
 
233
  # cache list of speakers and times they spoke
234
  self.speakers={}