yusenthebot commited on
Commit
716bda8
·
verified ·
1 Parent(s): 900b8d8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -5
requirements.txt CHANGED
@@ -1,6 +1,23 @@
 
1
  autogluon.tabular>=1.4.0
2
- gradio>=4.0.0
3
- huggingface_hub>=0.20.0
4
- pandas>=2.0.0
5
- scikit-learn>=1.0.0
6
- lightgbm>=3.0.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # --- Core ML stack ---
2
  autogluon.tabular>=1.4.0
3
+
4
+ # Tree models used by AutoGluon
5
+ lightgbm>=4.3.0
6
+
7
+ # --- FastAI (required because your trained ensemble includes NeuralNetFastAI) ---
8
+ fastai>=2.7,<3
9
+
10
+ # PyTorch CPU wheels (PIP 默认会装 CPU 版)
11
+ torch>=2.1,<3
12
+ torchvision>=0.16,<0.17
13
+ torchaudio>=2.1,<3
14
+
15
+ # --- Data / utils ---
16
+ pandas>=2.1
17
+ numpy>=1.24,<2.1
18
+ scikit-learn>=1.4
19
+
20
+ # --- Serving / Hub ---
21
+ gradio>=4.44.1
22
+ huggingface_hub>=0.25
23
+ fsspec>=2024.2