How to wrap AI models into containers