Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import os
 from langchain_groq import ChatGroq
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_community.embeddings import HuggingFaceBgeEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain_core.prompts import ChatPromptTemplate
@@ -10,6 +10,9 @@ from langchain_community.vectorstores import FAISS
 from langchain_community.document_loaders import PyPDFDirectoryLoader
 import time
 
+from dotenv import load_dotenv
+load_dotenv()
+
 # Retrieve API keys from environment variables
 huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 groq_api_key = os.getenv("GROQ_API_KEY")
@@ -45,7 +48,9 @@ Questions: {input}
 
 def vector_embedding():
     if "vectors" not in st.session_state:
-        st.session_state.embeddings = HuggingFaceEmbeddings()
+        st.session_state.embeddings=HuggingFaceBgeEmbeddings(model_name="BAAI/bge-small-en-v1.5",
+                                                            model_kwargs={'device':'cpu'},
+                                                            encode_kwargs={'normalize_embeddings':True})
         st.session_state.loader = PyPDFDirectoryLoader("./Data_Science")  # Data Ingestion
         st.session_state.docs = st.session_state.loader.load()  # Document Loading
         st.session_state.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)  # Chunk Creation