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