Update app.py

app.py

@@ -281,7 +281,7 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
     return chunks
 
 # Setup Chroma
-def setup_chroma(chunks, model_name="sentence-transformers/all-mpnet-base-v2", persist_directory="chroma_data"):
+def setup_chroma(chunks, model_name=model_name, persist_directory=persist_directory):
     embedding_model = HuggingFaceEmbeddings(model_name=model_name)
     vectorstore = Chroma.from_documents(chunks, embedding=embedding_model, persist_directory=persist_directory)
     return vectorstore