List of things to be done

1. is the base image correct or we can use a lighter image
2. mlflow or mlflow extras and need to be versioned