Spaces:

avid-ml
/

indie-label

Sleeping

indie-label / cache_model.py

update google drive link for new data folder

edd4636 over 2 years ago

284 Bytes

""""
Intended 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)

	""""
	Intended 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)