Astral.AI / requirements.txt
SilverDragon9's picture
Update requirements.txt
cebe18c verified
# ==============================
# 🎨 Gradio Interface
# ==============================
gradio>=4.0 # Web UI framework for interactive demos
Pillow>=10.0.0 # Image support (required by Gradio)
# ==============================
# ⚙️ Model Loading & Utilities
# ==============================
joblib>=1.3.0 # Serialization for trained ML models
numpy>=1.24.0 # Core numerical computing
scipy>=1.10.0 # Scientific computation support
# ==============================
# 🧠 Machine Learning (Tree-Based)
# ==============================
scikit-learn>=1.3.0 # ML model training, pipelines, preprocessing
xgboost>=1.7.5 # Advanced gradient boosting model
lightgbm>=4.1.0 # Lightweight boosting, optimized for speed
# catboost>=1.2.2 # Optional: GPU-optimized tree model with native categorical support
# ==============================
# 🧪 Experiment Tracking / Logging
# ==============================
rich>=13.0 # Pretty console output for logs
loguru>=0.7.0 # Elegant logging for development
tqdm>=4.66.0 # Loop progress bars
# ==============================
# 📊 Visualization & Debugging
# ==============================
matplotlib>=3.8.0 # Data and evaluation plots
seaborn>=0.13.0 # Statistical charts
plotly>=5.18.0 # Interactive, web-ready visualizations
# ==============================
# 🧰 File Management / API Utils
# ==============================
python-dotenv>=1.0.0 # .env file handling
requests>=2.31.0 # HTTP requests (e.g., APIs, model endpoints)
# ==============================
# ⚠️ Optional: Deep Learning / Transformers
# ==============================
# torch>=2.1.0 # Deep learning framework
# transformers>=4.36.0 # Hugging Face NLP models
# accelerate>=0.24.0 # Multi-GPU/TPU acceleration
# diffusers>=0.21.0 # Stable diffusion for image generation
# tensorflow>=2.15.0 # TensorFlow support
# keras>=2.15.0 # High-level API for deep learning
# ==============================
# 🧼 Code Quality / Dev Tools (Optional)
# ==============================
# black>=24.3.0 # Code formatter
# mypy>=1.8.0 # Static type checker