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

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

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

export class ClassificationModule extends BaseModule {
  protected static override nativeModule = ClassificationNativeModule;

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

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