FatimeFellowship / README.md
ghees's picture
Create README.md
5e233f8

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]