Update app.py

app.py

@@ -24,8 +24,6 @@ from itertools import combinations
 import pypdf
 import requests
 
-from chromadb.utils import embedding_functions
-from chromadb import Documents, EmbeddingFunction, Embeddings
 
 # LLM: openai and google_genai
 import openai
@@ -87,13 +85,6 @@ cohere_api_key = os.environ['cohere_api']
 
 current_dir = os.getcwd()
 
-# for new Chromadb   
-class MyEmbeddingFunction(EmbeddingFunction[Documents]):
-    def __call__(self, input: Documents) -> Embeddings:
-        sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="BAAI/bge-large-en-v1.5")
-        embeddings = sentence_transformer_ef(input)
-        return embeddings
-custom = MyEmbeddingFunction()
 
 
 
@@ -255,8 +246,7 @@ embeddings_HuggingFace = select_embeddings_model(LLM_service="HuggingFace")
 def create_vectorstore(embeddings,documents,vectorstore_name):
     """Create a Chroma vector database."""
     persist_directory = (current_dir + "/" + vectorstore_name)
-    #embedding_function=embeddings
-    embedding_function=custom
+    embedding_function=embeddings
     vector_store = Chroma.from_documents(
         documents=documents,
         embedding=embeddings,