GirishaBuilds01 commited on
Commit
f283da3
·
verified ·
1 Parent(s): 37156ae

Update embeddings.py

Browse files
Files changed (1) hide show
  1. embeddings.py +5 -3
embeddings.py CHANGED
@@ -1,7 +1,9 @@
1
- from sentence_transformers import SentenceTransformer
2
 
3
- model = SentenceTransformer("all-MiniLM-L6-v2")
4
 
5
  def embed(text):
6
 
7
- return model.encode(text).tolist()
 
 
 
1
+ from sklearn.feature_extraction.text import TfidfVectorizer
2
 
3
+ vectorizer = TfidfVectorizer(max_features=512)
4
 
5
  def embed(text):
6
 
7
+ vec = vectorizer.fit_transform([text])
8
+
9
+ return vec.toarray()[0].tolist()