Update appChatbot.py

appChatbot.py

@@ -7,6 +7,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 #from chromadb.utils import embedding_functions
+from langchain_community.embeddings import SentenceTransformerEmbeddings
 
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.embeddings import OpenAIEmbeddings
@@ -50,7 +51,8 @@ def init_chromadb():
     text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
     texts = text_splitter.split_documents(documents)
     # Select which embeddings we want to use
-    embeddings = OpenAIEmbeddings()
+    #embeddings = OpenAIEmbeddings()
+    embeddings = SentenceTransformerEmbeddings(model_name="nomic-ai/nomic-embed-text-v1", model_kwargs={"trust_remote_code":True}) 
     #query_chromadb()
 
     # Create the vectorestore to use as the index
@@ -64,7 +66,8 @@ def query_chromadb(ASK):
         raise Exception(f"{DB_DIR} does not exist, nothing can be queried")
 
     # Select which embeddings we want to use
-    embeddings = OpenAIEmbeddings()
+    #embeddings = OpenAIEmbeddings()
+    embeddings = SentenceTransformerEmbeddings(model_name="nomic-ai/nomic-embed-text-v1", model_kwargs={"trust_remote_code":True}) 
     # Load Vector store from local disk
     vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)