Updated

src/embedding/embedding.py

@@ -13,12 +13,14 @@ from langchain_huggingface import HuggingFaceEmbeddings
 #     return embedding_instance
 
 
-from langchain_huggingface import HuggingFaceEmbeddings
+from sentence_transformers import SentenceTransformer
+from langchain.embeddings import HuggingFaceEmbeddings
+
+# Load the sentence-transformers model directly
+model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 
-embedding_instance = HuggingFaceEmbeddings(
-    model_name="sentence-transformers/all-MiniLM-L6-v2",
-    model_kwargs={"device": "cpu"},
-)
+# Wrap it in LangChain HuggingFaceEmbeddings by passing the model instance
+embedding_instance = HuggingFaceEmbeddings(model=model)
 
 def all_minilm_l6_v2():
     """