Spaces:

kevinhug
/

ai

Sleeping

kevinhug commited on Apr 7

Commit

0f2b5ae

1 Parent(s): c613c64

local embedding

Files changed (1) hide show

rag.py CHANGED Viewed

@@ -20,7 +20,7 @@ from sentence_transformers import SentenceTransformer
 model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", device="cpu")
 embedder = dspy.Embedder(model.encode)
 """
-embedder = dspy.Embedder('huggingface/BAAI/bge-small-en')
 class RecommendProduct(dspy.Signature):
   """
   Recommend RBC financial product based on verbatim

 model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", device="cpu")
 embedder = dspy.Embedder(model.encode)
 """
+embedder = dspy.Embedder('huggingface/BAAI/bge-small-en-v1.5')
 class RecommendProduct(dspy.Signature):
   """
   Recommend RBC financial product based on verbatim