def __init__(self, model_path): self.model = load_model(model_path) self.uses_hist_equalization = get_hdf5_attr( model_path, 'decoder_uses_hist_equalization', True) self.distribution = DistributionCollection.from_hdf5(model_path) self._predict = predict_wrapper(self.model.predict, self.model.output_names) self.model._make_predict_function()