captain-awesome commited on
Commit
80e84cb
·
1 Parent(s): 2bd63e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -9,6 +9,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
9
  from langchain.text_splitter import RecursiveCharacterTextSplitter
10
  from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
11
  from langchain.vectorstores import Chroma
 
12
  from langchain.document_loaders import (
13
  CSVLoader,
14
  DirectoryLoader,
@@ -160,9 +161,11 @@ def create_vector_database(loaded_documents):
160
  chunked_documents = text_splitter.split_documents(loaded_documents)
161
 
162
  # Initialize HuggingFace embeddings
163
- embeddings = HuggingFaceEmbeddings(
164
- model_name="sentence-transformers/all-MiniLM-L6-v2"
165
- # model_name = "BAAI/bge-large-en"
 
 
166
  )
167
 
168
  # Create and persist a Chroma vector database from the chunked documents
 
9
  from langchain.text_splitter import RecursiveCharacterTextSplitter
10
  from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
11
  from langchain.vectorstores import Chroma
12
+ from langchain.embeddings import HuggingFaceBgeEmbeddings
13
  from langchain.document_loaders import (
14
  CSVLoader,
15
  DirectoryLoader,
 
161
  chunked_documents = text_splitter.split_documents(loaded_documents)
162
 
163
  # Initialize HuggingFace embeddings
164
+ # embeddings = HuggingFaceEmbeddings(
165
+ # model_name="sentence-transformers/all-MiniLM-L6-v2"
166
+ # )
167
+ embeddings = HuggingFaceBgeEmbeddings(
168
+ model_name = "BAAI/bge-large-en"
169
  )
170
 
171
  # Create and persist a Chroma vector database from the chunked documents