react-native-executorch/src/hooks/natural_language_processing/useTextEmbeddings.ts

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

import { TextEmbeddingsModule } from '../../modules/natural_language_processing/TextEmbeddingsModule';
import { ResourceSource } from '../../types/common';
import { useModule } from '../useModule';

export const useTextEmbeddings = ({
  modelSource,
  tokenizerSource,
  preventLoad = false,
}: {
  modelSource: ResourceSource;
  tokenizerSource: ResourceSource;
  preventLoad?: boolean;
}) =>
  useModule({
    module: TextEmbeddingsModule,
    loadArgs: [modelSource, tokenizerSource],
    preventLoad,
  });