zademn/mnist-mlops-learning

In this project I played with mlflow, streamlit and fastapi to create a training and prediction app on digits

Fastapi + MLflow + streamlit

Setup env. I hope I covered all.

pip install -r requirements.txt

Start app

Go in the root dir and run these

Streamlit

streamlit run frontend/streamlit_main.py

FastAPI

uvicorn backend.main:app

MLflow UI

mlflow ui --backend-store-uri sqlite:///db/bakckend.db

Docker

• Mlflow: http://localhost:5000
• FastApi: http://localhost:8000/docs
• Streamlit: http://localhost:8501/

docker-compose build
docker-compose up

Architecture

UI

TODO

• Dockerize
• Testing
• Maybe add celery instead of that background task? (Needs extra configs though)