diff --git a/mediapipe/tasks/python/text/__init__.py b/mediapipe/tasks/python/text/__init__.py index 5aa221c33..66c62cafc 100644 --- a/mediapipe/tasks/python/text/__init__.py +++ b/mediapipe/tasks/python/text/__init__.py @@ -14,9 +14,13 @@ """MediaPipe Tasks Text API.""" +import mediapipe.tasks.python.text.language_detector import mediapipe.tasks.python.text.text_classifier import mediapipe.tasks.python.text.text_embedder +LanguageDetector = language_detector.LanguageDetector +LanguageDetectorOptions = language_detector.LanguageDetectorOptions +LanguageDetectorResult = language_detector.LanguageDetectorResult TextClassifier = text_classifier.TextClassifier TextClassifierOptions = text_classifier.TextClassifierOptions TextClassifierResult = text_classifier.TextClassifierResult @@ -26,5 +30,6 @@ TextEmbedderResult = text_embedder.TextEmbedderResult # Remove unnecessary modules to avoid duplication in API docs. del mediapipe +del language_detector del text_classifier del text_embedder diff --git a/mediapipe/tasks/python/vision/__init__.py b/mediapipe/tasks/python/vision/__init__.py index 3c3f34db0..c88dbb9ad 100644 --- a/mediapipe/tasks/python/vision/__init__.py +++ b/mediapipe/tasks/python/vision/__init__.py @@ -15,6 +15,7 @@ """MediaPipe Tasks Vision API.""" import mediapipe.tasks.python.vision.core +import mediapipe.tasks.python.vision.face_aligner import mediapipe.tasks.python.vision.face_detector import mediapipe.tasks.python.vision.face_landmarker import mediapipe.tasks.python.vision.face_stylizer @@ -27,6 +28,8 @@ import mediapipe.tasks.python.vision.interactive_segmenter import mediapipe.tasks.python.vision.object_detector import mediapipe.tasks.python.vision.pose_landmarker +FaceAligner = face_aligner.FaceAligner +FaceAlignerOptions = face_aligner.FaceAlignerOptions FaceDetector = face_detector.FaceDetector FaceDetectorOptions = face_detector.FaceDetectorOptions FaceDetectorResult = face_detector.FaceDetectorResult @@ -65,6 +68,7 @@ RunningMode = core.vision_task_running_mode.VisionTaskRunningMode # Remove unnecessary modules to avoid duplication in API docs. del core +del face_aligner del face_detector del face_landmarker del face_stylizer