/** * Returns a TensorFlow model loader based on a serialized TensorFlow {@link Graph}. * * @param supplier {@link TensorFlowGraphModel} supplier. */ static TensorFlowGraphLoader create(final ThrowableSupplier<TensorFlowGraphModel> supplier) { final ModelLoader<TensorFlowGraphModel> loader = ModelLoader .lift(supplier) .with(ModelMemoizer::memoize) .with(Preloader.preloadAsync()); return loader::get; }
/** * Returns a TensorFlow model loader based on a saved model. * * @param supplier {@link TensorFlowModel} supplier. */ static TensorFlowLoader create(final ThrowableSupplier<TensorFlowModel> supplier) { final ModelLoader<TensorFlowModel> loader = ModelLoader .lift(supplier) .with(ModelMemoizer::memoize) .with(Preloader.preloadAsync()); return loader::get; }