def load(self, base):
    h5_path = glob.glob(os.path.join(base, '*.h5'))[0]
    self.model = utils.load_keras_model(h5_path)
def load(self, base):
    h5_path = glob.glob(os.path.join(base, '*.h5'))[0]
    self.model = utils.load_keras_model(h5_path)