A full-stack Machine Learning application that allows users to predict 8 different diseases based on user inputs. The system is designed with a modular backend in FastAPI and a responsive frontend in React with TailwindCSS, and includes functionality for online learning via user feedback and retraining!
- π©Ί Predicts the likelihood of:
- Anemia
- Cardiovascular Disease
- Heart Disease
- Hepatitis C
- Liver Disease
- Lung Cancer
- Stroke
- Thyroid Disease
- β‘ Built with FastAPI backend and Vite+React frontend
- π Supports user feedback and model retraining via
/submit_data/{disease}and/retrain/{disease}APIs - π§ͺ MLflow integration for experiment tracking
- π³ Dockerized for seamless deployment
| Layer | Technologies |
|---|---|
| Frontend | |
| Backend | |
| ML Models | |
| MLOps | |
| DevOps |
multi-disease-predictor/
βββ backend/ # FastAPI server + ML models + retraining
βββ frontend/ # React (Vite + TailwindCSS) UI
βββ models/ # Pretrained ML models
βββ notebooks/ # Jupyter Notebooks for experiments
βββ mlruns/ # MLflow runs
βββ mlartifacts/ # MLflow artifacts
βββ data/ # Raw and processed datasets
βββ docs/ # Sphinx documentation
βββ docker-compose.yml # Docker orchestration
βββ tests/ # API test scriptsπ§ Prerequisites
Python 3.10+
Node.js 18+
Docker (for containerized deployment)
pip or conda
#Create virtualenv and activate
python3 -m venv venv
source venv/bin/activatepip install -r requirements.txtcd backend
uvicorn main_2:app --reloadcd frontend
npm install
npm run devdocker-compose up --buildBackend: http://localhost:8000
Frontend: http://localhost:3000
Thanks to these amazing people :
| Name | GitHub Profile |
|---|---|
| Rohit Kshirsagar | @rohitkshirsagar19 |
| Parth Lhase | @LhaseParth2610 |
| Rishabh Kothari | @RishabhK103 |
| Prajwal Kumbhar | @prajwalkumbhar29 |
This project is licensed under the MIT License. See the LICENSE file for details.