maryzhang commited on
Commit
38fcc7d
·
verified ·
1 Parent(s): cfa1695

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -5
requirements.txt CHANGED
@@ -1,14 +1,17 @@
1
- # Core
2
  gradio>=4.0.0
3
  pandas>=2.0.0
4
- pillow>=9.5.0
5
  huggingface_hub>=0.20.0
6
 
7
- # AutoGluon (multimodal + tabular + core)
8
- autogluon.multimodal>=1.1.0
9
  autogluon.tabular>=1.1.0
10
  autogluon.core>=1.1.0
11
 
12
- # Dependencies required by trained models
13
  lightgbm>=4.0.0
 
 
 
 
 
14
  scikit-learn==1.6.1
 
1
+ # Core dependencies
2
  gradio>=4.0.0
3
  pandas>=2.0.0
 
4
  huggingface_hub>=0.20.0
5
 
6
+ # AutoGluon Tabular (ensemble models)
 
7
  autogluon.tabular>=1.1.0
8
  autogluon.core>=1.1.0
9
 
10
+ # Model backends used in ensembles
11
  lightgbm>=4.0.0
12
+ xgboost>=2.0.0
13
+ catboost>=1.2.0
14
+ fastai>=2.7.13
15
+
16
+ # Match training environment to avoid unpickling errors
17
  scikit-learn==1.6.1