Update requirements.txt

requirements.txt

@@ -2,62 +2,55 @@
 # 🎨 Gradio Interface
 # ==============================
 gradio>=4.0              # Web UI framework for interactive demos
-Pillow>=10.0.0           # Image processing for handling outputs
+Pillow>=10.0.0           # Image support (required by Gradio)
 
 # ==============================
 # ⚙️ Model Loading & Utilities
 # ==============================
-joblib>=1.3.0            # Efficient serialization & deserialization of ML models
-numpy>=1.24.0            # Core numerical computations
-scipy>=1.10.0            # Additional scientific 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 (General)
+# 🧠 Machine Learning (Tree-Based)
 # ==============================
-# scikit-learn>=1.3.0      # Tabular ML models and preprocessing utilities
-# xgboost>=1.7.5           # Gradient boosting models (popular for structured data)
-# lightgbm>=4.1.0          # Lightweight boosting framework optimized for performance
-
-# ==============================
-# 🔥 Deep Learning (Flexible)
-# ==============================
-# PyTorch Ecosystem
-torch>=2.1.0             # Deep learning framework
-torchvision>=0.16.0      # Image models, utilities, and datasets
-torchaudio>=2.1.0        # Audio processing with PyTorch
-
-# Hugging Face Transformers
-transformers>=4.36.0     # State-of-the-art NLP & vision transformer models
-diffusers>=0.21.0        # Stable diffusion for text-to-image generation
-accelerate>=0.24.0       # Hardware-aware training/inference for HF models
-
-# TensorFlow / Keras (Optional)
-# tensorflow>=2.15.0     # Google’s deep learning framework
-# keras>=2.15.0          # High-level API built on TensorFlow
+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 logging & formatted console output
-loguru>=0.7.0            # Simplified structured logging
-tqdm>=4.66.0             # Progress bars for loops and model runs
+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        # Basic visualization support
-seaborn>=0.13.0          # Statistical visualization
-plotly>=5.18.0           # Interactive plotting
-tensorboard>=2.15.0      # TensorFlow and PyTorch logging dashboard
+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     # Load env variables from .env files
-requests>=2.31.0         # HTTP calls for API/model interaction
+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 / Type Checking (Optional Dev Tools)
+# 🧼 Code Quality / Dev Tools (Optional)
 # ==============================
-# black>=24.3.0          # Auto-formatter (commented for production)
-# mypy>=1.8.0            # Type checker for Python (optional for robustness)
+# black>=24.3.0            # Code formatter
+# mypy>=1.8.0              # Static type checker