Update app.py

app.py

@@ -56,6 +56,7 @@ import gradio as gr
 import tempfile
 import timeit
 import textwrap
+from chromadb.utils import embedding_functions
 
 FILE_LOADER_MAPPING = {
     "csv": (CSVLoader, {"encoding": "utf-8"}),
@@ -108,10 +109,11 @@ def create_vector_database(loaded_documents):
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=30, length_function = len)
     chunked_documents = text_splitter.split_documents(loaded_documents)
 
-    embeddings = HuggingFaceEmbeddings(
-        model_name="sentence-transformers/all-MiniLM-L6-v2"
-        # model_name = "sentence-transformers/all-mpnet-base-v2"
-    )
+    # embeddings = HuggingFaceEmbeddings(
+    #     model_name="sentence-transformers/all-MiniLM-L6-v2"
+    #     # model_name = "sentence-transformers/all-mpnet-base-v2"
+    # )
+    embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="all-MiniLM-L6-v2")
 
     # embeddings = HuggingFaceBgeEmbeddings(
     #     model_name = "BAAI/bge-large-en"