Update rag.py

rag.py

@@ -10,6 +10,14 @@ from langchain.vectorstores.utils import filter_complex_metadata
 #add new import
 from langchain_community.document_loaders.csv_loader import CSVLoader
 
+from sentence_transformers import SentenceTransformer
+
+from langchain_community.embeddings import HuggingFaceEmbeddings
+model_name = "sentence-transformers/all-mpnet-base-v2"
+embedding = HuggingFaceEmbeddings(
+    model_name=model_name,
+   )
+
 
 
 class ChatPDF:
@@ -38,7 +46,7 @@ class ChatPDF:
         chunks = self.text_splitter.split_documents(docs)
         chunks = filter_complex_metadata(chunks)
 
-        vector_store = Chroma.from_documents(documents=chunks, embedding=FastEmbedEmbeddings())
+        vector_store = Chroma.from_documents(documents=chunks, embedding=embedding)
         self.retriever = vector_store.as_retriever(
             search_type="similarity_score_threshold",
             search_kwargs={