react-native-executorch/src/modules/computer_vision/ObjectDetectionModule.ts

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

import { ObjectDetectionNativeModule } from '../../native/RnExecutorchModules';
import { ResourceSource } from '../../types/common';
import { BaseModule } from '../BaseModule';

export class ObjectDetectionModule extends BaseModule {
  protected static override nativeModule = ObjectDetectionNativeModule;

  static override async load(modelSource: ResourceSource) {
    return await super.load(modelSource);
  }

  static override async forward(
    input: string
  ): ReturnType<typeof this.nativeModule.forward> {
    return await this.nativeModule.forward(input);
  }
}