futranbg commited on
Commit
f0779a5
·
verified ·
1 Parent(s): 32da345

Create download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +4 -0
download_model.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from sentence_transformers import SentenceTransformer
2
+ import os
3
+ model_name = os.getenv("MODEL_NAME", "dangvantuan/vietnamese-embedding")
4
+ model = SentenceTransformer(model_name)