react-native-executorch/lib/modules/general/ExecutorchModule.js

An easy way to run AI models in React Native with ExecuTorch

import { BaseNonStaticModule } from '../BaseNonStaticModule';
import { ResourceFetcher } from '../../utils/ResourceFetcher';
export class ExecutorchModule extends BaseNonStaticModule {
    async load(modelSource, onDownloadProgressCallback = () => { }) {
        const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, modelSource);
        if (paths === null || paths.length < 1) {
            throw new Error('Download interrupted.');
        }
        this.nativeModule = global.loadExecutorchModule(paths[0] || '');
    }
    async forward(inputTensor) {
        return await this.forwardET(inputTensor);
    }
}