bupa1018 commited on
Commit
81004d4
·
1 Parent(s): b1b2c63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -281,7 +281,7 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
281
  return chunks
282
 
283
  # Setup Chroma
284
- def setup_chroma(chunks, model_name=model_name, persist_directory=persist_directory):
285
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
286
  vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
287
  return vectorstore
 
281
  return chunks
282
 
283
  # Setup Chroma
284
+ def setup_chroma(chunks, model_name, persist_directory):
285
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
286
  vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
287
  return vectorstore