Deployment Code for model deployment A simple example of setting up train scheduler and convert the model into API using FastAPI