File size: 805 Bytes
f58e385 eb66dcb f58e385 eb66dcb f58e385 eb66dcb f58e385 |
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) |