react-native-executorch/src/hooks/computer_vision/useImageEmbeddings.ts

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

import { ImageEmbeddingsModule } from '../../modules/computer_vision/ImageEmbeddingsModule';
import { ResourceSource } from '../../types/common';
import { useModule } from '../useModule';

interface Props {
  model: { modelSource: ResourceSource };
  preventLoad?: boolean;
}

export const useImageEmbeddings = ({ model, preventLoad = false }: Props) =>
  useModule({
    module: ImageEmbeddingsModule,
    model,
    preventLoad,
  });