Sediba-AI commited on
Commit ·
f4b8ea1
1
Parent(s): 9e7e9eb
feat: add model download script for HF Space startup
Browse files- download_models.py +15 -0
download_models.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Run at Space startup to pull models from HF Hub."""
|
| 2 |
+
from huggingface_hub import snapshot_download
|
| 3 |
+
import os
|
| 4 |
+
|
| 5 |
+
os.makedirs("models/zabantu_nso_sentiment", exist_ok=True)
|
| 6 |
+
|
| 7 |
+
if not os.path.exists("models/zabantu_nso_sentiment/config.json"):
|
| 8 |
+
print("Downloading zabantu-nso-sentiment...")
|
| 9 |
+
snapshot_download(
|
| 10 |
+
repo_id="Sediba-AI/zabantu-nso-sentiment",
|
| 11 |
+
local_dir="models/zabantu_nso_sentiment"
|
| 12 |
+
)
|
| 13 |
+
print("Done.")
|
| 14 |
+
else:
|
| 15 |
+
print("Model already present.")
|