Change typing in useInstanceSegmentation

Author: benITo47

packages/react-native-executorch/src/hooks/computer_vision/useInstanceSegmentation.ts

@@ -5,7 +5,6 @@ import {
 import {
   InstanceSegmentationProps,
   InstanceSegmentationType,
-  InstanceModelNameOf,
   InstanceSegmentationModelSources,
 } from '../../types/instanceSegmentation';
 import { useModuleFactory } from '../useModuleFactory';
@@ -54,16 +53,17 @@ export const useInstanceSegmentation = <
     downloadProgress,
     runForward,
     instance,
-  } = useModuleFactory<InstanceSegmentationModule<InstanceModelNameOf<C>>, C>({
-    factory: InstanceSegmentationModule.fromModelName,
+  } = useModuleFactory({
+    factory: (config, onProgress) =>
+      InstanceSegmentationModule.fromModelName(config, onProgress),
     config: model,
     preventLoad,
   });
 
   const forward: InstanceSegmentationType<
     InstanceSegmentationLabels<C['modelName']>
   >['forward'] = (imageSource, options) =>
-    runForward((instance) => instance.forward(imageSource, options) as any);
+    runForward((inst) => inst.forward(imageSource, options));
 
   const getAvailableInputSizes = () => instance?.getAvailableInputSizes();
 

packages/react-native-executorch/src/modules/computer_vision/InstanceSegmentationModule.ts

@@ -88,7 +88,7 @@ type ModelConfigsType = typeof ModelConfigs;
  */
 export type InstanceSegmentationLabels<
   M extends InstanceSegmentationModelName,
-> = ModelConfigsType[M]['labelMap'];
+> = ResolveLabels<M>;
 
 /**
  * @internal