RAG / embedding_function.py
GLITCH-08's picture
Update embedding_function.py
1e295d6 verified
Raw
History Blame Contribute Delete
247 Bytes
from langchain.embeddings import HuggingFaceEmbeddings
def get_embedding():
# Use a Hugging Face-hosted model that outputs 768-dimensional embeddings
embeddings = HuggingFaceEmbeddings(model_name="bert-base-uncased")
return embeddings