Preprocessing before feeding to model
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('paraphrase-MiniLM-L6-v2', device='cuda')
...
embeddings = model.encode([text])
return embeddings[0]
Preprocessing before feeding to model
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('paraphrase-MiniLM-L6-v2', device='cuda')
...
embeddings = model.encode([text])
return embeddings[0]