genaibeauty commited on
Commit
ffe3a85
Β·
verified Β·
1 Parent(s): 98d9183

Create embeddings.py

Browse files
Files changed (1) hide show
  1. embeddings.py +8 -0
embeddings.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from langchain.embeddings import HuggingFaceEmbeddings
2
+
3
+ # Load Hugging Face model for embedding generation
4
+ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
5
+
6
+ def get_embedding(text):
7
+ """Generate embeddings for a given text."""
8
+ return embedding_model.embed_query(text)