Update app.py

app.py

@@ -430,6 +430,8 @@ from langchain_community.embeddings import SentenceTransformerEmbeddings
 from langchain_community.llms import HuggingFacePipeline
 from langchain.chains import RetrievalQA
 from langchain.schema import Document
+from sentence_transformers import SentenceTransformer
+from langchain_community.embeddings import HuggingFaceEmbeddings
 
 # --- Configuration ---
 st.set_page_config(page_title="📚 RAG PDF Chatbot", layout="wide")
@@ -468,7 +470,8 @@ def split_text_into_chunks(text):
 
 # --- Create Vector DB ---
 def create_vectorstore(documents):
-    embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+    model = SentenceTransformer("all-MiniLM-L6-v2", device='cpu')
+    embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
     db = Chroma.from_documents(documents, embeddings, persist_directory=persist_directory)
     db.persist()
     return db