bupa1018 commited on
Commit
616f4b7
·
1 Parent(s): eafcd92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -282,7 +282,7 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
282
  return chunks
283
 
284
  # Setup Vectorstore
285
- def setup_vectorstore(chunks, model_name):
286
  print("Start setup_vectorstore_function")
287
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
288
 
@@ -386,7 +386,7 @@ def initialize():
386
  #print(f"Total number of code_chunks: {len(code_chunks)}")
387
  print(f"Total number of doc_chunks: {len(doc_chunks)}")
388
 
389
- docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME)
390
  #codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
391
 
392
  #llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
 
282
  return chunks
283
 
284
  # Setup Vectorstore
285
+ def setup_vectorstore(chunks, model_name, persist_directory):
286
  print("Start setup_vectorstore_function")
287
  embedding_model = HuggingFaceEmbeddings(model_name=model_name)
288
 
 
386
  #print(f"Total number of code_chunks: {len(code_chunks)}")
387
  print(f"Total number of doc_chunks: {len(doc_chunks)}")
388
 
389
+ docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
390
  #codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
391
 
392
  #llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)