wayne-chi commited on
Commit
5bb82f3
·
verified ·
1 Parent(s): 8a5686e

Upload 3 files

Browse files
Files changed (3) hide show
  1. download_models.py +28 -0
  2. requirements.txt +21 -3
  3. setup.sh +2 -0
download_models.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from huggingface_hub import hf_hub_download, snapshot_download
3
+
4
+ # Target directory for models
5
+ target_dir = "Models"
6
+ os.makedirs(target_dir, exist_ok=True)
7
+
8
+ # Download specific files (Folds 1–5) from willieseun/Eagle-Team-TabPFN
9
+ print("Downloading fold models from willieseun/Eagle-Team-TabPFN...")
10
+ for i in range(1, 6):
11
+ file_name = f"Fold_{i}_best_model.tabpfn_fit"
12
+ model_path = hf_hub_download(
13
+ repo_id="willieseun/Eagle-Team-TabPFN",
14
+ filename=file_name,
15
+ local_dir=target_dir
16
+ )
17
+ print(f"Downloaded: {model_path}")
18
+
19
+ # Download full snapshot from wayne-chi/Eagle_Team
20
+ print("\nDownloading snapshot from wayne-chi/Eagle_Team...")
21
+ snapshot_download(
22
+ repo_id="wayne-chi/Eagle_Team",
23
+ revision="main", # Optional, default is "main"
24
+ local_dir=target_dir,
25
+ local_dir_use_symlinks=False
26
+ )
27
+
28
+ print("\n✅ All models downloaded successfully to:", target_dir)
requirements.txt CHANGED
@@ -1,3 +1,21 @@
1
- altair
2
- pandas
3
- streamlit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ tabpfn-extensions @ git+https://github.com/PriorLabs/tabpfn-extensions.git@16e0e4f4305a3546eab5be6ebf163ff41bd3843d
2
+ scikit-learn==1.5.1
3
+ huggingface_hub
4
+ autogluon
5
+ tabpfn==2.0.9
6
+ streamlit==1.43.0
7
+ numpy==1.26.4
8
+ pandas==2.2.3
9
+ matplotlib
10
+ matplotlib-inline==0.1.7
11
+ seaborn==0.13.2
12
+ torch
13
+ setuptools
14
+ boto3==1.40.2
15
+ accelerate==1.9.0
16
+ certifi==2025.7.14
17
+ catboost==1.2.8
18
+ charset-normalizer==3.4.2
19
+ coreforecast==0.0.16
20
+ fastai==2.7.19
21
+
setup.sh ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ #!/bin/bash
2
+ python download_models.py