martina3435 commited on
Commit
7df2e0f
·
verified ·
1 Parent(s): 8a7a2a4

Update generate_embeddings.py

Browse files
Files changed (1) hide show
  1. generate_embeddings.py +1 -1
generate_embeddings.py CHANGED
@@ -36,7 +36,7 @@ def split_documents(docs):
36
 
37
  def create_vectorstore(docs):
38
  embedding = HuggingFaceEmbeddings(
39
- model_name="sentence-transformers/all-MiniLM-L6-v2"
40
  )
41
 
42
  print("Creating vector DB...")
 
36
 
37
  def create_vectorstore(docs):
38
  embedding = HuggingFaceEmbeddings(
39
+ model_name="BAAI/bge-base-en-v1.5"
40
  )
41
 
42
  print("Creating vector DB...")