Yeah this was partially the purpose of the Loader class which ModelRunner/ImageClassifier/TextTranslator derive from. Perhaps more things could be pulled into Loader, or into Caffe2ModelLoader/ONNXModelLoader. If you have any ideas we always welcome PRs