From d7a4680d17ec465751f5c5df472b58626b2d2171 Mon Sep 17 00:00:00 2001 From: Derek Mauro Date: Fri, 14 Oct 2022 14:33:26 -0700 Subject: [PATCH] No description PiperOrigin-RevId: 481235204 --- .../classification_by_retrieval/lib/model_builder.cc | 6 ++++-- .../classification_by_retrieval/lib/model_builder.h | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/lite/examples/classification_by_retrieval/lib/model_builder.cc b/lite/examples/classification_by_retrieval/lib/model_builder.cc index f3879722f9d..dc74d623a8a 100644 --- a/lite/examples/classification_by_retrieval/lib/model_builder.cc +++ b/lite/examples/classification_by_retrieval/lib/model_builder.cc @@ -1,5 +1,7 @@ #include "lib/model_builder.h" +#include + #include "absl/status/status.h" #include "absl/strings/str_format.h" #include "absl/strings/str_join.h" @@ -61,8 +63,8 @@ ModelBuilder::CreateFromImageEmbedderOptions( return absl::InvalidArgumentError("Failed to build model from file: " + model_file); } - return absl::make_unique(std::move(image_embedder), - std::move(model)); + return std::make_unique(std::move(image_embedder), + std::move(model)); } void ModelBuilder::SetMetadata( diff --git a/lite/examples/classification_by_retrieval/lib/model_builder.h b/lite/examples/classification_by_retrieval/lib/model_builder.h index 8a3b48c08b9..a8f7d32fbe6 100644 --- a/lite/examples/classification_by_retrieval/lib/model_builder.h +++ b/lite/examples/classification_by_retrieval/lib/model_builder.h @@ -43,7 +43,7 @@ class ModelBuilder { std::unique_ptr<::tflite::task::vision::ImageEmbedder> image_embedder, std::unique_ptr<::tflite::FlatBufferModel> model, std::unique_ptr tflite_cbr_builder = - absl::make_unique()); + std::make_unique()); // Initializes the ModelBuilder from the provided ImageEmbedderOptions. static tflite::support::StatusOr>