WebashalarForML commited on
Commit
2fa40d0
·
verified ·
1 Parent(s): 3898411

Update retrival.py

Browse files
Files changed (1) hide show
  1. retrival.py +2 -2
retrival.py CHANGED
@@ -271,8 +271,8 @@ def save_to_chroma(chunks: list[Document], name: str):
271
 
272
  try:
273
  # Load the embedding model
274
- #embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
275
- embedding_function = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1")
276
  # Create Chroma DB for documents using from_documents [NOTE: Some of the data is converted to string because int and float show null if added]
277
  print("Creating document vector database...")
278
  db = Chroma.from_documents(
 
271
 
272
  try:
273
  # Load the embedding model
274
+ embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
275
+ #embedding_function = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1")
276
  # Create Chroma DB for documents using from_documents [NOTE: Some of the data is converted to string because int and float show null if added]
277
  print("Creating document vector database...")
278
  db = Chroma.from_documents(