react-native-executorch/lib/module/hooks/computer_vision/useStyleTransfer.js

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

"use strict";

import { useState } from 'react';
import { _StyleTransferModule } from '../../native/RnExecutorchModules';
import { useModule } from '../useModule';
export const useStyleTransfer = ({
  modelSource
}) => {
  const [module, _] = useState(() => new _StyleTransferModule());
  const {
    error,
    isReady,
    isGenerating,
    downloadProgress,
    forwardImage: forward
  } = useModule({
    modelSource,
    module
  });
  return {
    error,
    isReady,
    isGenerating,
    downloadProgress,
    forward
  };
};
//# sourceMappingURL=useStyleTransfer.js.map