Sediba-AI commited on
Commit
f4b8ea1
·
1 Parent(s): 9e7e9eb

feat: add model download script for HF Space startup

Browse files
Files changed (1) hide show
  1. 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.")