Trying to fix BAAI/BGE

app_final.py

@@ -26,7 +26,7 @@ from langchain_core.runnables import RunnablePassthrough, RunnableLambda, Runnab
 from langchain_core.runnables import chain
 
 import langchain_community.embeddings.huggingface
-from langchain_community.embeddings.huggingface import HuggingFaceBgeEmbeddings, HuggingFaceEmbeddings
+from langchain_community.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 
 from langchain.chains import LLMChain
@@ -91,7 +91,7 @@ model_name = "BAAI/bge-large-en-v1.5"
 model_kwargs = {"device": "cpu"}
 encode_kwargs = {"normalize_embeddings": True}
 if "embeddings" not in st.session_state:
-    st.session_state.embeddings = HuggingFaceBgeEmbeddings(
+    st.session_state.embeddings = HuggingFaceEmbeddings(
         model_name=model_name, 
         model_kwargs = model_kwargs,
         encode_kwargs = encode_kwargs)