Update embeddings.py

embeddings.py

@@ -1,8 +1,8 @@
-from langchain.embeddings import HuggingFaceEmbeddings
+from sentence_transformers import SentenceTransformer
 
-# Load Hugging Face model for embedding generation
-embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+# Load the pre-trained SentenceTransformer model
+model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 
 def get_embedding(text):
-    """Generate embeddings for a given text."""
-    return embedding_model.embed_query(text)
+    """Generate embeddings for a given text using SentenceTransformer."""
+    return model.encode([text])[0]  # Return the first embedding (only one sentence)