HateBot is a Dockerized pipeline to train, track, and deploy a hate speech detection model. The system is fully automated with CI/CD using GitHub Actions and deploys the model to a private server via SSH.
Features
- 🧠 Model Implementation with PyTorch and Transformers library
- 🎯 Hyperparameter Optimization using Optuna
- 📊 ML Lifecycle Management via MLflow
- 🚀 Model Serving using FastAPI
- 🐳 Dockerized for pipeline for portability and reproducibility
- 🔁 CI/CD with GitHub Actions
- 📦 Deployment over SSH to self-hosted server