saherPervaiz commited on
Commit
f91e02f
·
verified ·
1 Parent(s): 6bd6ab4

Update embedder.py

Browse files
Files changed (1) hide show
  1. embedder.py +3 -5
embedder.py CHANGED
@@ -1,8 +1,6 @@
1
- # embedder.py
2
  from sentence_transformers import SentenceTransformer
3
 
4
- # Load transformer model once
5
- tf_model = SentenceTransformer("all-MiniLM-L6-v2")
6
 
7
- def get_embeddings(text_list):
8
- return tf_model.encode(text_list, convert_to_numpy=True)
 
 
1
  from sentence_transformers import SentenceTransformer
2
 
3
+ model = SentenceTransformer("all-MiniLM-L6-v2")
 
4
 
5
+ def get_embeddings(texts):
6
+ return model.encode(texts)