react-native-executorch
Version:
An easy way to run AI models in react native with ExecuTorch
17 lines • 605 B
TypeScript
import { ETInput, Module } from '../types/common';
import { _ETModule } from '../native/RnExecutorchModules';
interface Props {
modelSource: string | number;
module: Module;
}
interface _Module {
error: string | null;
isReady: boolean;
isGenerating: boolean;
downloadProgress: number;
forwardETInput: (input: ETInput[] | ETInput, shape: number[][] | number[]) => ReturnType<_ETModule['forward']>;
forwardImage: (input: string) => Promise<any>;
}
export declare const useModule: ({ modelSource, module }: Props) => _Module;
export {};
//# sourceMappingURL=useModule.d.ts.map