talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

No public description

Browse Source 
PiperOrigin-RevId: 563543098
This commit is contained in:
MediaPipe Team
2023-09-07 14:19:08 -07:00
committed by Copybara-Service
parent 6c38483b37
commit 80b762a281
5 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/c/text/text_classifier/BUILD
+1
View File
@@ -26,5 +26,6 @@ cc_library(
"//mediapipe/tasks/c/components/processors:classifier_options",
"//mediapipe/tasks/c/core:base_options",
"//mediapipe/tasks/cc/text/text_classifier",
"@com_google_absl//absl/log:absl_log",
],
)
mediapipe/tasks/c/text/text_classifier/text_classifier.cc
+5 -3
View File
@@ -17,6 +17,7 @@ limitations under the License.
#include <memory>
#include "absl/log/absl_log.h"
#include "mediapipe/tasks/c/components/containers/classification_result.h"
#include "mediapipe/tasks/c/components/processors/classifier_options.h"
#include "mediapipe/tasks/c/core/base_options.h"
@@ -44,7 +45,8 @@ TextClassifier* CppTextClassifierCreate(TextClassifierOptions options) {
auto classifier = TextClassifier::Create(std::move(cpp_options));
if (!classifier.ok()) {
LOG(ERROR) << "Failed to create TextClassifier: " << classifier.status();
ABSL_LOG(ERROR) << "Failed to create TextClassifier: "
<< classifier.status();
return nullptr;
}
return classifier->release();
@@ -55,7 +57,7 @@ bool CppTextClassifierClassify(void* classifier, char* utf8_str,
auto cpp_classifier = static_cast<TextClassifier*>(classifier);
auto cpp_result = cpp_classifier->Classify(utf8_str);
if (!cpp_result.ok()) {
LOG(ERROR) << "Classification failed: " << cpp_result.status();
ABSL_LOG(ERROR) << "Classification failed: " << cpp_result.status();
return false;
}
CppConvertToClassificationResult(*cpp_result, result);
@@ -66,7 +68,7 @@ void CppTextClassifierClose(void* classifier) {
auto cpp_classifier = static_cast<TextClassifier*>(classifier);
auto result = cpp_classifier->Close();
if (!result.ok()) {
LOG(ERROR) << "Failed to close TextClassifier: " << result;
ABSL_LOG(ERROR) << "Failed to close TextClassifier: " << result;
}
delete cpp_classifier;
}