react-native-executorch/src/native/NativeImageSegmentation.ts

An easy way to run AI models in react native with ExecuTorch

import type { TurboModule } from 'react-native';
import { TurboModuleRegistry } from 'react-native';

export interface Spec extends TurboModule {
  loadModule(modelSource: string): Promise<number>;

  forward(
    input: string,
    classesOfInterest: string[],
    resize: boolean
  ): Promise<{ [category: string]: number[] }>;
}

export default TurboModuleRegistry.get<Spec>('ImageSegmentation');