Rohit-Katkar2003 commited on
Commit
16c0f06
Β·
verified Β·
1 Parent(s): 98e8511

update model file

Browse files
Files changed (1) hide show
  1. download_model.py +13 -12
download_model.py CHANGED
@@ -1,13 +1,14 @@
1
- import os
2
- from huggingface_hub import snapshot_download
3
-
4
- MODEL_REPO = "facebook/mobilellm-pro" # ← YOUR model
5
- LOCAL_DIR = "/app/model"
6
-
7
- print(f"πŸ“₯ Downloading model from {MODEL_REPO}...")
8
- snapshot_download(
9
- repo_id=MODEL_REPO,
10
- local_dir=LOCAL_DIR,
11
- local_dir_use_symlinks=False # avoids symlink issues in Docker
12
- )
 
13
  print("βœ… Model downloaded successfully!")
 
1
+ import os
2
+ from huggingface_hub import snapshot_download
3
+ hf_token = os.getenv("HF_TOKEN", None)
4
+
5
+ MODEL_REPO = "facebook/mobilellm-pro" # ← YOUR model
6
+ LOCAL_DIR = "/app/model"
7
+
8
+ print(f"πŸ“₯ Downloading model from {MODEL_REPO}...")
9
+ snapshot_download(
10
+ repo_id=MODEL_REPO,
11
+ local_dir=LOCAL_DIR,
12
+ local_dir_use_symlinks=False # avoids symlink issues in Docker
13
+ )
14
  print("βœ… Model downloaded successfully!")