File size: 763 Bytes
eb66dcb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from sentence_transformers import SentenceTransformer
class GetEmbedding:
def __init__(self,data:list):
self.data = data
def user_query_emb(self,model_name:str = 'paraphrase-MiniLM-L6-v2'):
try:
model = SentenceTransformer(model_name_or_path=model_name)
embedding = model.encode(self.data)
return embedding
except Exception as e:
print(e)
def convert_data(self,model_name:str = 'paraphrase-MiniLM-L6-v2'):
try:
model = SentenceTransformer(model_name)
embeddings = model.encode(self.data)
return embeddings
except Exception as e:
print(e)
if __name__ == "__main__":
emb = GetEmbedding("lalit")
print( emb) |