Update llm.py

llm.py

@@ -5,7 +5,7 @@ from langchain.vectorstores import Chroma
 from langchain_core.prompts import ChatPromptTemplate
 from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain.chains import create_retrieval_chain
-from langchain_community.embeddings import HuggingFaceBgeEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings
 import pandas as pd
 import os
 from langchain.schema import Document
@@ -27,9 +27,9 @@ def load_and_process_pdf():
     return splits
 
 def create_vectorstore(splits = load_and_process_pdf()):
-    model_name = "nomic-ai/nomic-embed-text-v1"
-    embeddings = HuggingFaceBgeEmbeddings(
-        model_name=model_name)
+    # model_name = "nomic-ai/nomic-embed-text-v1"
+    embeddings = HuggingFaceEmbedding()
+    #     model_name=model_name)
     vectorstore = Chroma(
         collection_name="example_collection",
         embedding_function=embeddings,