Use model bundle writer when exporting models in gesture recognizer

PiperOrigin-RevId: 487042776
2022-11-08 13:52:08 -08:00
parent 0363d60511
commit b3d19fa1af
8 changed files with 138 additions and 9 deletions
@@ -144,8 +144,13 @@ filegroup(
 )

 # Gestures related models. Visible to model_maker.
+# TODO: Upload canned gesture model and gesture embedding model to GCS after Model Card approval
 filegroup(
    name = "test_gesture_models",
+    srcs = [
+        "hand_landmark_full.tflite",
+        "palm_detection_full.tflite",
+    ],
    visibility = [
        "//mediapipe/model_maker:__subpackages__",
        "//mediapipe/tasks:internal",