NDMO_Assistant / embedding.py
MOHAMMED-N's picture
Upload 6 files
0de70e8 verified
raw
history blame contribute delete
337 Bytes
# embedding.py
from langchain_huggingface import HuggingFaceEmbeddings
def load_embeddings():
"""
Returns a HuggingFaceEmbeddings instance.
"""
embeddings = HuggingFaceEmbeddings(
model_name="CAMeL-Lab/bert-base-arabic-camelbert-mix",
model_kwargs={"trust_remote_code": True}
)
return embeddings