Skip to content

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

Notifications You must be signed in to change notification settings

suhyun6363/mnist-mlops-learning

 
 

Repository files navigation

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/backend.db

Docker

docker-compose build
docker-compose up

Architecture

image

UI

image image

TODO

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 94.9%
  • Dockerfile 5.1%