🐋 Writing predict py for a Dockerized ML Model