bupa1018 commited on
Commit
094f515
·
1 Parent(s): 7f71568

Create embeddings.py

Browse files
Files changed (1) hide show
  1. embeddings.py +12 -0
embeddings.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.embeddings import HuggingFaceEmbeddings
2
+
3
+ def get_hf_embeddings(model_name=None):
4
+ """Retrieve a Hugging Face embedding model using the specified model name.."""
5
+
6
+ if model_name is None:
7
+ # "sentence-transformers/all-mpnet-base-v2"
8
+ model_name = "BAAI/bge-base-en-v1.5"
9
+
10
+ embeddings = HuggingFaceEmbeddings(model_name=model_name)
11
+
12
+ return embeddings