book-recommendation-predictor / requirements.txt
yusenthebot's picture
Update requirements.txt
716bda8 verified
raw
history blame
451 Bytes
# --- Core ML stack ---
autogluon.tabular>=1.4.0
# Tree models used by AutoGluon
lightgbm>=4.3.0
# --- FastAI (required because your trained ensemble includes NeuralNetFastAI) ---
fastai>=2.7,<3
# PyTorch CPU wheels (PIP 默认会装 CPU 版)
torch>=2.1,<3
torchvision>=0.16,<0.17
torchaudio>=2.1,<3
# --- Data / utils ---
pandas>=2.1
numpy>=1.24,<2.1
scikit-learn>=1.4
# --- Serving / Hub ---
gradio>=4.44.1
huggingface_hub>=0.25
fsspec>=2024.2