update bge recipe

bedapudi6788 · bedapudi6788 · commit 80a04004fcf2 · 2024-10-29T14:15:22.000+05:30
Signed-off-by: Praneeth Bedapudi &lt;praneeth@bpraneeth.com&gt;
diff --git a/recipes/text_embeddings/predictor.py b/recipes/text_embeddings/predictor.py
@@ -2,7 +2,7 @@
 
 sentences = ['That is a happy person', 'That is a very happy person']
 
-model = SentenceTransformer('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True)
+model = SentenceTransformer('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True, backend="onnx", model_kwargs={"file_name": "model.onnx", "provider": "CPUExecutionProvider"})
 
 from time import time
 
diff --git a/recipes/text_embeddings/requirements.txt b/recipes/text_embeddings/requirements.txt
@@ -1 +1 @@
-sentence-transformers
+sentence-transformers[onnx]
