Spaces:
Running
Running
Update download_model.py
Browse files- download_model.py +3 -1
download_model.py
CHANGED
|
@@ -4,7 +4,9 @@ Script สำหรับ download model ล่วงหน้าตอน Docke
|
|
| 4 |
"""
|
| 5 |
#login huggingface
|
| 6 |
from huggingface_hub import login
|
| 7 |
-
|
|
|
|
|
|
|
| 8 |
|
| 9 |
#download model
|
| 10 |
from sentence_transformers import SentenceTransformer
|
|
|
|
| 4 |
"""
|
| 5 |
#login huggingface
|
| 6 |
from huggingface_hub import login
|
| 7 |
+
import os
|
| 8 |
+
|
| 9 |
+
login(token=os.getenv("HF_TOKEN"))
|
| 10 |
|
| 11 |
#download model
|
| 12 |
from sentence_transformers import SentenceTransformer
|