Save class names in checkpoints and update the high-level inference APIs (#645)

* update the high-level inference api

* save classes in meta data and use it for visualization