Added prevent load to all hooks

Author: pweglik

src/hooks/computer_vision/useClassification.ts

@@ -4,6 +4,13 @@ import { useModule } from '../useModule';
 
 export const useClassification = ({
   modelSource,
+  preventLoad = false,
 }: {
   modelSource: ResourceSource;
-}) => useModule({ module: ClassificationModule, loadArgs: [modelSource] });
+  preventLoad?: boolean;
+}) =>
+  useModule({
+    module: ClassificationModule,
+    loadArgs: [modelSource],
+    preventLoad,
+  });

src/hooks/computer_vision/useImageSegmentation.ts

@@ -1,9 +1,18 @@
 import { useModule } from '../useModule';
 import { ImageSegmentationModule } from '../../modules/computer_vision/ImageSegmentationModule';
+import { ResourceSource } from '../../types/common';
 
 interface Props {
-  modelSource: string | number;
+  modelSource: ResourceSource;
+  preventLoad?: boolean;
 }
 
-export const useImageSegmentation = ({ modelSource }: Props) =>
-  useModule({ module: ImageSegmentationModule, loadArgs: [modelSource] });
+export const useImageSegmentation = ({
+  modelSource,
+  preventLoad = false,
+}: Props) =>
+  useModule({
+    module: ImageSegmentationModule,
+    loadArgs: [modelSource],
+    preventLoad,
+  });

src/hooks/computer_vision/useOCR.ts

@@ -15,6 +15,7 @@ export const useOCR = ({
   detectorSource,
   recognizerSources,
   language = 'en',
+  preventLoad = false,
 }: {
   detectorSource: ResourceSource;
   recognizerSources: {
@@ -23,6 +24,7 @@ export const useOCR = ({
     recognizerSmall: ResourceSource;
   };
   language?: OCRLanguage;
+  preventLoad?: boolean;
 }): OCRModule => {
   const [error, setError] = useState<string | null>(null);
   const [isReady, setIsReady] = useState(false);
@@ -45,9 +47,18 @@ export const useOCR = ({
       await model.loadModel(detectorSource, recognizerSources, language);
     };
 
-    loadModel();
+    if (!preventLoad) {
+      loadModel();
+    }
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [model, detectorSource, language, JSON.stringify(recognizerSources)]);
+  }, [
+    model,
+    detectorSource,
+    language,
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    JSON.stringify(recognizerSources),
+    preventLoad,
+  ]);
 
   return {
     error,

src/hooks/computer_vision/useObjectDetection.ts

@@ -4,7 +4,15 @@ import { ObjectDetectionModule } from '../../modules/computer_vision/ObjectDetec
 
 interface Props {
   modelSource: ResourceSource;
+  preventLoad?: boolean;
 }
 
-export const useObjectDetection = ({ modelSource }: Props) =>
-  useModule({ module: ObjectDetectionModule, loadArgs: [modelSource] });
+export const useObjectDetection = ({
+  modelSource,
+  preventLoad = false,
+}: Props) =>
+  useModule({
+    module: ObjectDetectionModule,
+    loadArgs: [modelSource],
+    preventLoad,
+  });

src/hooks/computer_vision/useStyleTransfer.ts

@@ -4,7 +4,12 @@ import { StyleTransferModule } from '../../modules/computer_vision/StyleTransfer
 
 interface Props {
   modelSource: ResourceSource;
+  preventLoad?: boolean;
 }
 
-export const useStyleTransfer = ({ modelSource }: Props) =>
-  useModule({ module: StyleTransferModule, loadArgs: [modelSource] });
+export const useStyleTransfer = ({ modelSource, preventLoad = false }: Props) =>
+  useModule({
+    module: StyleTransferModule,
+    loadArgs: [modelSource],
+    preventLoad,
+  });

src/hooks/computer_vision/useVerticalOCR.ts

@@ -16,6 +16,7 @@ export const useVerticalOCR = ({
   recognizerSources,
   language = 'en',
   independentCharacters = false,
+  preventLoad = false,
 }: {
   detectorSources: {
     detectorLarge: ResourceSource;
@@ -27,6 +28,7 @@ export const useVerticalOCR = ({
   };
   language?: OCRLanguage;
   independentCharacters?: boolean;
+  preventLoad?: boolean;
 }): OCRModule => {
   const [error, setError] = useState<string | null>(null);
   const [isReady, setIsReady] = useState(false);
@@ -54,7 +56,9 @@ export const useVerticalOCR = ({
       );
     };
 
-    loadModel();
+    if (!preventLoad) {
+      loadModel();
+    }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [
     model,
@@ -64,6 +68,7 @@ export const useVerticalOCR = ({
     independentCharacters,
     // eslint-disable-next-line react-hooks/exhaustive-deps
     JSON.stringify(recognizerSources),
+    preventLoad,
   ]);
 
   return {

src/hooks/general/useExecutorchModule.ts

@@ -1,9 +1,18 @@
 import { ExecutorchModule } from '../../modules/general/ExecutorchModule';
+import { ResourceSource } from '../../types/common';
 import { useModule } from '../useModule';
 
 interface Props {
-  modelSource: string | number;
+  modelSource: ResourceSource;
+  preventLoad?: boolean;
 }
 
-export const useExecutorchModule = ({ modelSource }: Props) =>
-  useModule({ module: ExecutorchModule, loadArgs: [modelSource] });
+export const useExecutorchModule = ({
+  modelSource,
+  preventLoad = false,
+}: Props) =>
+  useModule({
+    module: ExecutorchModule,
+    loadArgs: [modelSource],
+    preventLoad,
+  });

src/hooks/natural_language_processing/useSpeechToText.ts

@@ -30,6 +30,7 @@ export const useSpeechToText = ({
   overlapSeconds,
   windowSize,
   streamingConfig,
+  preventLoad = false,
 }: {
   modelName: AvailableModels;
   encoderSource?: ResourceSource;
@@ -44,6 +45,7 @@ export const useSpeechToText = ({
   streamingConfig?: ConstructorParameters<
     typeof SpeechToTextController
   >['0']['streamingConfig'];
+  preventLoad?: boolean;
 }): SpeechToTextModule => {
   const [sequence, setSequence] = useState<string>('');
   const [isReady, setIsReady] = useState(false);
@@ -76,8 +78,17 @@ export const useSpeechToText = ({
         tokenizerSource
       );
     };
-    loadModel();
-  }, [model, modelName, encoderSource, decoderSource, tokenizerSource]);
+    if (!preventLoad) {
+      loadModel();
+    }
+  }, [
+    model,
+    modelName,
+    encoderSource,
+    decoderSource,
+    tokenizerSource,
+    preventLoad,
+  ]);
 
   return {
     isReady,

src/hooks/natural_language_processing/useTextEmbeddings.ts

@@ -5,11 +5,14 @@ import { useModule } from '../useModule';
 export const useTextEmbeddings = ({
   modelSource,
   tokenizerSource,
+  preventLoad = false,
 }: {
   modelSource: ResourceSource;
   tokenizerSource: ResourceSource;
+  preventLoad?: boolean;
 }) =>
   useModule({
     module: TextEmbeddingsModule,
     loadArgs: [modelSource, tokenizerSource],
+    preventLoad,
   });

src/hooks/natural_language_processing/useTokenizer.ts

@@ -5,8 +5,10 @@ import { ETError, getError } from '../../Error';
 
 export const useTokenizer = ({
   tokenizerSource,
+  preventLoad = false,
 }: {
   tokenizerSource: ResourceSource;
+  preventLoad?: boolean;
 }) => {
   const [error, setError] = useState<null | string>(null);
   const [isReady, setIsReady] = useState(false);
@@ -24,8 +26,10 @@ export const useTokenizer = ({
         setError((err as Error).message);
       }
     };
-    loadModule();
-  }, [tokenizerSource]);
+    if (!preventLoad) {
+      loadModule();
+    }
+  }, [tokenizerSource, preventLoad]);
 
   const stateWrapper = <T extends (...args: any[]) => Promise<any>>(fn: T) => {
     const boundFn = fn.bind(TokenizerModule);