File size: 284 Bytes
0362c93
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
""""
Intented for use by Dockerfile; downloads pytorch model and caches it 
in the image, so the download doesn't have to run when app starts.
"""

from sentence_transformers import SentenceTransformer

model_name = "paraphrase-MiniLM-L6-v2"
model = SentenceTransformer(model_name)