aboutsummaryrefslogtreecommitdiff
path: root/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h')
-rw-r--r--tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h7
1 files changed, 6 insertions, 1 deletions
diff --git a/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h b/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h
index 013b7d53..8e70ba4c 100644
--- a/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h
+++ b/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h
@@ -109,9 +109,14 @@ class NLClassifier : public core::BaseTaskApi<std::vector<core::Category>,
std::unique_ptr<tflite::OpResolver> resolver =
absl::make_unique<tflite::ops::builtin::BuiltinOpResolver>());
- // Performs classification on a string input, returns classified results.
+ // DEPRECATED (unannotated for backward compatibility). Prefer using `ClassifyText`.
std::vector<core::Category> Classify(const std::string& text);
+ // Performs classification on a string input, returns classified results or an
+ // error.
+ tflite::support::StatusOr<std::vector<core::Category>> ClassifyText(
+ const std::string& text);
+
// Gets the model version, or "NO_VERSION_INFO" in case there is no version.
std::string GetModelVersion() const;