local embedding
Browse files
rag.py
CHANGED
|
@@ -20,7 +20,7 @@ from sentence_transformers import SentenceTransformer
|
|
| 20 |
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", device="cpu")
|
| 21 |
embedder = dspy.Embedder(model.encode)
|
| 22 |
"""
|
| 23 |
-
embedder = dspy.Embedder('huggingface/BAAI/bge-small-en')
|
| 24 |
class RecommendProduct(dspy.Signature):
|
| 25 |
"""
|
| 26 |
Recommend RBC financial product based on verbatim
|
|
|
|
| 20 |
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", device="cpu")
|
| 21 |
embedder = dspy.Embedder(model.encode)
|
| 22 |
"""
|
| 23 |
+
embedder = dspy.Embedder('huggingface/BAAI/bge-small-en-v1.5')
|
| 24 |
class RecommendProduct(dspy.Signature):
|
| 25 |
"""
|
| 26 |
Recommend RBC financial product based on verbatim
|