Add prototype MediaPipe adapter to builtin Model Explorer adapters

PiperOrigin-RevId: 703671776
google-ai-edge · Dec 17, 2024 · 8690b21 · 8690b21
1 parent 2da380f
commit 8690b21
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/ui/src/common/types.ts b/src/ui/src/common/types.ts
@@ -36,6 +36,7 @@ export enum InternalAdapterExtId {
   MLIR = 'builtin_mlir',
   JSON_LOADER = 'builtin_json',
   DATA_NEXUS = 'builtin_data_nexus',
+  MEDIAPIPE = 'builtin_mediapipe',
 }
 
 /** Extension types. */

diff --git a/src/ui/src/services/internal_adapter_extensions.ts b/src/ui/src/services/internal_adapter_extensions.ts
@@ -88,6 +88,14 @@ const dataNexusAdapterExtension: AdapterExtension = {
   name: 'Data Nexus adapter',
   description: 'Loads data from Data Nexus.',
 };
+const mediapipeAdapterExtension: AdapterExtension = {
+  type: ExtensionType.ADAPTER,
+  fileExts: ['pbtxt'],
+  id: InternalAdapterExtId.MEDIAPIPE,
+  name: 'MediaPipe adapter',
+  description:
+    'A built-in adapter that converts a MediaPipe Pipeline to Model Explorer format.',
+};
 
 /** All internal extensions. */
 export const INTERNAL_ADAPTER_EXTENSIONS: AdapterExtension[] = [
@@ -99,4 +107,5 @@ export const INTERNAL_ADAPTER_EXTENSIONS: AdapterExtension[] = [
   mlirAdapterExtension,
   jsonAdapterExtension,
   dataNexusAdapterExtension,
+  mediapipeAdapterExtension,
 ];