react-native-executorch 0.5.1-rc.0 → 0.5.1

package/lib/typescript/hooks/natural_language_processing/useSpeechToText.d.ts

@@ -1,28 +1,19 @@
-import { SpeechToTextController } from '../../controllers/SpeechToTextController';
-import { ResourceSource } from '../../types/common';
-import { STREAMING_ACTION } from '../../constants/sttDefaults';
-import { AvailableModels, SpeechToTextLanguage } from '../../types/stt';
-interface SpeechToTextModule {
+import { SpeechToTextModelConfig } from '../../types/stt';
+export declare const useSpeechToText: ({ model, preventLoad, }: {
+    model: SpeechToTextModelConfig;
+    preventLoad?: boolean;
+}) => {
+    error: string | null;
     isReady: boolean;
     isGenerating: boolean;
-    sequence: string;
     downloadProgress: number;
-    configureStreaming: SpeechToTextController['configureStreaming'];
-    error: Error | undefined;
-    transcribe: (input: number[], audioLanguage?: SpeechToTextLanguage) => ReturnType<SpeechToTextController['transcribe']>;
-    streamingTranscribe: (streamAction: STREAMING_ACTION, input?: number[], audioLanguage?: SpeechToTextLanguage) => ReturnType<SpeechToTextController['streamingTranscribe']>;
-}
-export declare const useSpeechToText: ({ model, overlapSeconds, windowSize, streamingConfig, preventLoad, }: {
-    model: {
-        modelName: AvailableModels;
-        encoderSource: ResourceSource;
-        decoderSource: ResourceSource;
-        tokenizerSource: ResourceSource;
-    };
-    overlapSeconds?: ConstructorParameters<typeof SpeechToTextController>["0"]["overlapSeconds"];
-    windowSize?: ConstructorParameters<typeof SpeechToTextController>["0"]["windowSize"];
-    streamingConfig?: ConstructorParameters<typeof SpeechToTextController>["0"]["streamingConfig"];
-    preventLoad?: boolean;
-}) => SpeechToTextModule;
-export {};
+    committedTranscription: string;
+    nonCommittedTranscription: string;
+    encode: (waveform: Float32Array<ArrayBufferLike>) => Promise<void>;
+    decode: (tokens: number[]) => Promise<Float32Array<ArrayBufferLike>>;
+    transcribe: (waveform: number[], options?: import("../../types/stt").DecodingOptions | undefined) => Promise<string>;
+    stream: () => Promise<string>;
+    streamStop: () => void;
+    streamInsert: (waveform: number[]) => void;
+};
 //# sourceMappingURL=useSpeechToText.d.ts.map

package/lib/typescript/hooks/natural_language_processing/useSpeechToText.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useSpeechToText.d.ts","sourceRoot":"","sources":["../../../../src/hooks/natural_language_processing/useSpeechToText.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,0CAA0C,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAExE,UAAU,kBAAkB;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,EAAE,OAAO,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,gBAAgB,EAAE,MAAM,CAAC;IACzB,kBAAkB,EAAE,sBAAsB,CAAC,oBAAoB,CAAC,CAAC;IACjE,KAAK,EAAE,KAAK,GAAG,SAAS,CAAC;IACzB,UAAU,EAAE,CACV,KAAK,EAAE,MAAM,EAAE,EACf,aAAa,CAAC,EAAE,oBAAoB,KACjC,UAAU,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC,CAAC;IACtD,mBAAmB,EAAE,CACnB,YAAY,EAAE,gBAAgB,EAC9B,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,aAAa,CAAC,EAAE,oBAAoB,KACjC,UAAU,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC,CAAC;CAChE;AAED,eAAO,MAAM,eAAe,GAAI,sEAM7B;IACD,KAAK,EAAE;QACL,SAAS,EAAE,eAAe,CAAC;QAC3B,aAAa,EAAE,cAAc,CAAC;QAC9B,aAAa,EAAE,cAAc,CAAC;QAC9B,eAAe,EAAE,cAAc,CAAC;KACjC,CAAC;IACF,cAAc,CAAC,EAAE,qBAAqB,CACpC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC;IACzB,UAAU,CAAC,EAAE,qBAAqB,CAChC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;IACrB,eAAe,CAAC,EAAE,qBAAqB,CACrC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,iBAAiB,CAAC,CAAC;IAC1B,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,KAAG,kBAoEH,CAAC"}
+{"version":3,"file":"useSpeechToText.d.ts","sourceRoot":"","sources":["../../../../src/hooks/natural_language_processing/useSpeechToText.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAE1D,eAAO,MAAM,eAAe,GAAI,yBAG7B;IACD,KAAK,EAAE,uBAAuB,CAAC;IAC/B,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;;;;;;;;;;;;;CAmGA,CAAC"}

package/lib/typescript/index.d.ts

@@ -1,4 +1,3 @@
-import { SpeechToTextLanguage } from './types/stt';
 declare global {
     var loadStyleTransfer: (source: string) => any;
     var loadImageSegmentation: (source: string) => any;
@@ -42,9 +41,9 @@ export * from './types/objectDetection';
 export * from './types/ocr';
 export * from './types/imageSegmentation';
 export * from './types/llm';
-export { SpeechToTextLanguage };
+export * from './types/common';
+export { SpeechToTextLanguage, SpeechToTextModelConfig, DecodingOptions, } from './types/stt';
 export * from './constants/modelUrls';
 export * from './constants/ocr/models';
 export * from './constants/llmDefaults';
-export { STREAMING_ACTION, MODES, AvailableModels, } from './constants/sttDefaults';
 //# sourceMappingURL=index.d.ts.map

package/lib/typescript/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAKnD,OAAO,CAAC,MAAM,CAAC;IACb,IAAI,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAC/C,IAAI,qBAAqB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACnD,IAAI,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAChD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,oBAAoB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAClD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,kBAAkB,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,KAAK,GAAG,CAAC;IAC9E,IAAI,gBAAgB,EAAE,CACpB,aAAa,EAAE,MAAM,EACrB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,MAAM,KACd,GAAG,CAAC;IACT,IAAI,OAAO,EAAE,CACX,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,MAAM,EACvB,gBAAgB,EAAE,MAAM,EACxB,eAAe,EAAE,MAAM,EACvB,OAAO,EAAE,MAAM,KACZ,GAAG,CAAC;IACT,IAAI,eAAe,EAAE,CACnB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,qBAAqB,CAAC,EAAE,OAAO,KAC5B,GAAG,CAAC;CACV;AAyBD,cAAc,2CAA2C,CAAC;AAC1D,cAAc,4CAA4C,CAAC;AAC3D,cAAc,0CAA0C,CAAC;AACzD,cAAc,8CAA8C,CAAC;AAC7D,cAAc,gCAAgC,CAAC;AAC/C,cAAc,wCAAwC,CAAC;AACvD,cAAc,4CAA4C,CAAC;AAE3D,cAAc,4CAA4C,CAAC;AAC3D,cAAc,qDAAqD,CAAC;AACpE,cAAc,uDAAuD,CAAC;AACtE,cAAc,kDAAkD,CAAC;AAEjE,cAAc,qCAAqC,CAAC;AAGpD,cAAc,gDAAgD,CAAC;AAC/D,cAAc,iDAAiD,CAAC;AAChE,cAAc,+CAA+C,CAAC;AAC9D,cAAc,mDAAmD,CAAC;AAClE,cAAc,qCAAqC,CAAC;AACpD,cAAc,6CAA6C,CAAC;AAC5D,cAAc,oCAAoC,CAAC;AACnD,cAAc,iDAAiD,CAAC;AAEhE,cAAc,iDAAiD,CAAC;AAChE,cAAc,0DAA0D,CAAC;AACzE,cAAc,4DAA4D,CAAC;AAC3E,cAAc,uDAAuD,CAAC;AAGtE,cAAc,yBAAyB,CAAC;AACxC,cAAc,aAAa,CAAC;AAG5B,cAAc,yBAAyB,CAAC;AACxC,cAAc,aAAa,CAAC;AAC5B,cAAc,2BAA2B,CAAC;AAC1C,cAAc,aAAa,CAAC;AAC5B,OAAO,EAAE,oBAAoB,EAAE,CAAC;AAGhC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,gBAAgB,EAChB,KAAK,EACL,eAAe,GAChB,MAAM,yBAAyB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,CAAC,MAAM,CAAC;IACb,IAAI,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAC/C,IAAI,qBAAqB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACnD,IAAI,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAChD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,oBAAoB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IAClD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,mBAAmB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,GAAG,CAAC;IACjD,IAAI,kBAAkB,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,KAAK,GAAG,CAAC;IAC9E,IAAI,gBAAgB,EAAE,CACpB,aAAa,EAAE,MAAM,EACrB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,MAAM,KACd,GAAG,CAAC;IACT,IAAI,OAAO,EAAE,CACX,cAAc,EAAE,MAAM,EACtB,eAAe,EAAE,MAAM,EACvB,gBAAgB,EAAE,MAAM,EACxB,eAAe,EAAE,MAAM,EACvB,OAAO,EAAE,MAAM,KACZ,GAAG,CAAC;IACT,IAAI,eAAe,EAAE,CACnB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,qBAAqB,CAAC,EAAE,OAAO,KAC5B,GAAG,CAAC;CACV;AAyBD,cAAc,2CAA2C,CAAC;AAC1D,cAAc,4CAA4C,CAAC;AAC3D,cAAc,0CAA0C,CAAC;AACzD,cAAc,8CAA8C,CAAC;AAC7D,cAAc,gCAAgC,CAAC;AAC/C,cAAc,wCAAwC,CAAC;AACvD,cAAc,4CAA4C,CAAC;AAE3D,cAAc,4CAA4C,CAAC;AAC3D,cAAc,qDAAqD,CAAC;AACpE,cAAc,uDAAuD,CAAC;AACtE,cAAc,kDAAkD,CAAC;AAEjE,cAAc,qCAAqC,CAAC;AAGpD,cAAc,gDAAgD,CAAC;AAC/D,cAAc,iDAAiD,CAAC;AAChE,cAAc,+CAA+C,CAAC;AAC9D,cAAc,mDAAmD,CAAC;AAClE,cAAc,qCAAqC,CAAC;AACpD,cAAc,6CAA6C,CAAC;AAC5D,cAAc,oCAAoC,CAAC;AACnD,cAAc,iDAAiD,CAAC;AAEhE,cAAc,iDAAiD,CAAC;AAChE,cAAc,0DAA0D,CAAC;AACzE,cAAc,4DAA4D,CAAC;AAC3E,cAAc,uDAAuD,CAAC;AAGtE,cAAc,yBAAyB,CAAC;AACxC,cAAc,aAAa,CAAC;AAG5B,cAAc,yBAAyB,CAAC;AACxC,cAAc,aAAa,CAAC;AAC5B,cAAc,2BAA2B,CAAC;AAC1C,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,OAAO,EACL,oBAAoB,EACpB,uBAAuB,EACvB,eAAe,GAChB,MAAM,aAAa,CAAC;AAGrB,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,yBAAyB,CAAC"}

package/lib/typescript/modules/natural_language_processing/SpeechToTextModule.d.ts

@@ -1,25 +1,22 @@
-import { ResourceSource } from '../../types/common';
-import { SpeechToTextController } from '../../controllers/SpeechToTextController';
-import { AvailableModels, SpeechToTextLanguage } from '../../types/stt';
-import { STREAMING_ACTION } from '../../constants/sttDefaults';
+import { DecodingOptions, SpeechToTextModelConfig } from '../../types/stt';
 export declare class SpeechToTextModule {
-    private module;
-    constructor({ transcribeCallback, overlapSeconds, windowSize, streamingConfig, }?: {
-        transcribeCallback?: (sequence: string) => void;
-        overlapSeconds?: ConstructorParameters<typeof SpeechToTextController>['0']['overlapSeconds'];
-        windowSize?: ConstructorParameters<typeof SpeechToTextController>['0']['windowSize'];
-        streamingConfig?: ConstructorParameters<typeof SpeechToTextController>['0']['streamingConfig'];
-    });
-    load(model: {
-        modelName: AvailableModels;
-        encoderSource?: ResourceSource;
-        decoderSource?: ResourceSource;
-        tokenizerSource?: ResourceSource;
-    }, onDownloadProgressCallback?: (progress: number) => void): Promise<void>;
-    configureStreaming(overlapSeconds: Parameters<SpeechToTextController['configureStreaming']>[0], windowSize: Parameters<SpeechToTextController['configureStreaming']>[1], streamingConfig: Parameters<SpeechToTextController['configureStreaming']>[2]): void;
-    encode(waveform: Float32Array): Promise<null>;
-    decode(seq: number[]): Promise<number>;
-    transcribe(waveform: number[], audioLanguage?: SpeechToTextLanguage): ReturnType<SpeechToTextController['transcribe']>;
-    streamingTranscribe(streamAction: STREAMING_ACTION, waveform?: number[], audioLanguage?: SpeechToTextLanguage): ReturnType<SpeechToTextController['streamingTranscribe']>;
+    private modelConfig;
+    private asr;
+    private processor;
+    private isStreaming;
+    private readyToProcess;
+    private minAudioSamples;
+    load(model: SpeechToTextModelConfig, onDownloadProgressCallback?: (progress: number) => void): Promise<void>;
+    encode(waveform: Float32Array): Promise<void>;
+    decode(tokens: number[]): Promise<Float32Array>;
+    transcribe(waveform: number[], options?: DecodingOptions): Promise<string>;
+    stream(options?: DecodingOptions): AsyncGenerator<{
+        committed: string;
+        nonCommitted: string;
+    }, void, unknown>;
+    streamStop(): void;
+    streamInsert(waveform: number[]): void;
+    private validateOptions;
+    private resetStreamState;
 }
 //# sourceMappingURL=SpeechToTextModule.d.ts.map

package/lib/typescript/modules/natural_language_processing/SpeechToTextModule.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SpeechToTextModule.d.ts","sourceRoot":"","sources":["../../../../src/modules/natural_language_processing/SpeechToTextModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,0CAA0C,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAE/D,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,MAAM,CAAyB;gBAE3B,EACV,kBAAkB,EAClB,cAAc,EACd,UAAU,EACV,eAAe,GAChB,GAAE;QACD,kBAAkB,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;QAChD,cAAc,CAAC,EAAE,qBAAqB,CACpC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC;QACzB,UAAU,CAAC,EAAE,qBAAqB,CAChC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;QACrB,eAAe,CAAC,EAAE,qBAAqB,CACrC,OAAO,sBAAsB,CAC9B,CAAC,GAAG,CAAC,CAAC,iBAAiB,CAAC,CAAC;KACtB;IASA,IAAI,CACR,KAAK,EAAE;QACL,SAAS,EAAE,eAAe,CAAC;QAC3B,aAAa,CAAC,EAAE,cAAc,CAAC;QAC/B,aAAa,CAAC,EAAE,cAAc,CAAC;QAC/B,eAAe,CAAC,EAAE,cAAc,CAAC;KAClC,EACD,0BAA0B,GAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAe;IAWnE,kBAAkB,CAChB,cAAc,EAAE,UAAU,CAAC,sBAAsB,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,EAC3E,UAAU,EAAE,UAAU,CAAC,sBAAsB,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,EACvE,eAAe,EAAE,UAAU,CAAC,sBAAsB,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC;IAKxE,MAAM,CAAC,QAAQ,EAAE,YAAY;IAI7B,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE;IAIpB,UAAU,CACd,QAAQ,EAAE,MAAM,EAAE,EAClB,aAAa,CAAC,EAAE,oBAAoB,GACnC,UAAU,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;IAI7C,mBAAmB,CACvB,YAAY,EAAE,gBAAgB,EAC9B,QAAQ,CAAC,EAAE,MAAM,EAAE,EACnB,aAAa,CAAC,EAAE,oBAAoB,GACnC,UAAU,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC;CAO7D"}
+{"version":3,"file":"SpeechToTextModule.d.ts","sourceRoot":"","sources":["../../../../src/modules/natural_language_processing/SpeechToTextModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAI3E,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,GAAG,CAAkB;IAE7B,OAAO,CAAC,SAAS,CAAwD;IACzE,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,eAAe,CAAqB;IAE/B,IAAI,CACf,KAAK,EAAE,uBAAuB,EAC9B,0BAA0B,GAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAe;IAMtD,MAAM,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7C,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAI/C,UAAU,CACrB,QAAQ,EAAE,MAAM,EAAE,EAClB,OAAO,GAAE,eAAoB,GAC5B,OAAO,CAAC,MAAM,CAAC;IAeJ,MAAM,CAAC,OAAO,GAAE,eAAoB;;;;IA2B3C,UAAU;IAIV,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE;IAKtC,OAAO,CAAC,eAAe;IASvB,OAAO,CAAC,gBAAgB;CAKzB"}

package/lib/typescript/types/stt.d.ts

@@ -1,95 +1,21 @@
-export interface ModelConfig {
-    sources: {
-        encoder: string;
-        decoder: string;
-    };
-    tokenizer: {
-        source: string;
-        bos: number;
-        eos: number;
-    };
-    isMultilingual: boolean;
+import { ResourceSource } from './common';
+export type WordTuple = [number, number, string];
+export interface WordObject {
+    start: number;
+    end: number;
+    word: string;
+}
+export interface Segment {
+    words: WordObject[];
 }
-export declare enum SpeechToTextLanguage {
-    Afrikaans = "af",
-    Albanian = "sq",
-    Arabic = "ar",
-    Armenian = "hy",
-    Azerbaijani = "az",
-    Basque = "eu",
-    Belarusian = "be",
-    Bengali = "bn",
-    Bosnian = "bs",
-    Bulgarian = "bg",
-    Burmese = "my",
-    Catalan = "ca",
-    Chinese = "zh",
-    Croatian = "hr",
-    Czech = "cs",
-    Danish = "da",
-    Dutch = "nl",
-    Estonian = "et",
-    English = "en",
-    Finnish = "fi",
-    French = "fr",
-    Galician = "gl",
-    Georgian = "ka",
-    German = "de",
-    Greek = "el",
-    Gujarati = "gu",
-    HaitianCreole = "ht",
-    Hebrew = "he",
-    Hindi = "hi",
-    Hungarian = "hu",
-    Icelandic = "is",
-    Indonesian = "id",
-    Italian = "it",
-    Japanese = "ja",
-    Kannada = "kn",
-    Kazakh = "kk",
-    Khmer = "km",
-    Korean = "ko",
-    Lao = "lo",
-    Latvian = "lv",
-    Lithuanian = "lt",
-    Macedonian = "mk",
-    Malagasy = "mg",
-    Malay = "ms",
-    Malayalam = "ml",
-    Maltese = "mt",
-    Marathi = "mr",
-    Nepali = "ne",
-    Norwegian = "no",
-    Persian = "fa",
-    Polish = "pl",
-    Portuguese = "pt",
-    Punjabi = "pa",
-    Romanian = "ro",
-    Russian = "ru",
-    Serbian = "sr",
-    Sinhala = "si",
-    Slovak = "sk",
-    Slovenian = "sl",
-    Spanish = "es",
-    Sundanese = "su",
-    Swahili = "sw",
-    Swedish = "sv",
-    Tagalog = "tl",
-    Tajik = "tg",
-    Tamil = "ta",
-    Telugu = "te",
-    Thai = "th",
-    Turkish = "tr",
-    Ukrainian = "uk",
-    Urdu = "ur",
-    Uzbek = "uz",
-    Vietnamese = "vi",
-    Welsh = "cy",
-    Yiddish = "yi"
+export type SpeechToTextLanguage = 'af' | 'sq' | 'ar' | 'hy' | 'az' | 'eu' | 'be' | 'bn' | 'bs' | 'bg' | 'my' | 'ca' | 'zh' | 'hr' | 'cs' | 'da' | 'nl' | 'et' | 'en' | 'fi' | 'fr' | 'gl' | 'ka' | 'de' | 'el' | 'gu' | 'ht' | 'he' | 'hi' | 'hu' | 'is' | 'id' | 'it' | 'ja' | 'kn' | 'kk' | 'km' | 'ko' | 'lo' | 'lv' | 'lt' | 'mk' | 'mg' | 'ms' | 'ml' | 'mt' | 'mr' | 'ne' | 'no' | 'fa' | 'pl' | 'pt' | 'pa' | 'ro' | 'ru' | 'sr' | 'si' | 'sk' | 'sl' | 'es' | 'su' | 'sw' | 'sv' | 'tl' | 'tg' | 'ta' | 'te' | 'th' | 'tr' | 'uk' | 'ur' | 'uz' | 'vi' | 'cy' | 'yi';
+export interface DecodingOptions {
+    language?: SpeechToTextLanguage;
 }
-export declare enum AvailableModels {
-    WHISPER = "whisper",
-    MOONSHINE = "moonshine",
-    WHISPER_MULTILINGUAL = "whisperMultilingual"
+export interface SpeechToTextModelConfig {
+    isMultilingual: boolean;
+    encoderSource: ResourceSource;
+    decoderSource: ResourceSource;
+    tokenizerSource: ResourceSource;
 }
 //# sourceMappingURL=stt.d.ts.map

package/lib/typescript/types/stt.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../../../src/types/stt.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,SAAS,EAAE;QACT,MAAM,EAAE,MAAM,CAAC;QACf,GAAG,EAAE,MAAM,CAAC;QACZ,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,cAAc,EAAE,OAAO,CAAC;CACzB;AAGD,oBAAY,oBAAoB;IAC9B,SAAS,OAAO;IAChB,QAAQ,OAAO;IACf,MAAM,OAAO;IACb,QAAQ,OAAO;IACf,WAAW,OAAO;IAClB,MAAM,OAAO;IACb,UAAU,OAAO;IACjB,OAAO,OAAO;IACd,OAAO,OAAO;IACd,SAAS,OAAO;IAChB,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,QAAQ,OAAO;IACf,KAAK,OAAO;IACZ,MAAM,OAAO;IACb,KAAK,OAAO;IACZ,QAAQ,OAAO;IACf,OAAO,OAAO;IACd,OAAO,OAAO;IACd,MAAM,OAAO;IACb,QAAQ,OAAO;IACf,QAAQ,OAAO;IACf,MAAM,OAAO;IACb,KAAK,OAAO;IACZ,QAAQ,OAAO;IACf,aAAa,OAAO;IACpB,MAAM,OAAO;IACb,KAAK,OAAO;IACZ,SAAS,OAAO;IAChB,SAAS,OAAO;IAChB,UAAU,OAAO;IACjB,OAAO,OAAO;IACd,QAAQ,OAAO;IACf,OAAO,OAAO;IACd,MAAM,OAAO;IACb,KAAK,OAAO;IACZ,MAAM,OAAO;IACb,GAAG,OAAO;IACV,OAAO,OAAO;IACd,UAAU,OAAO;IACjB,UAAU,OAAO;IACjB,QAAQ,OAAO;IACf,KAAK,OAAO;IACZ,SAAS,OAAO;IAChB,OAAO,OAAO;IACd,OAAO,OAAO;IACd,MAAM,OAAO;IACb,SAAS,OAAO;IAChB,OAAO,OAAO;IACd,MAAM,OAAO;IACb,UAAU,OAAO;IACjB,OAAO,OAAO;IACd,QAAQ,OAAO;IACf,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,MAAM,OAAO;IACb,SAAS,OAAO;IAChB,OAAO,OAAO;IACd,SAAS,OAAO;IAChB,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,KAAK,OAAO;IACZ,KAAK,OAAO;IACZ,MAAM,OAAO;IACb,IAAI,OAAO;IACX,OAAO,OAAO;IACd,SAAS,OAAO;IAChB,IAAI,OAAO;IACX,KAAK,OAAO;IACZ,UAAU,OAAO;IACjB,KAAK,OAAO;IACZ,OAAO,OAAO;CACf;AAED,oBAAY,eAAe;IACzB,OAAO,YAAY;IACnB,SAAS,cAAc;IACvB,oBAAoB,wBAAwB;CAC7C"}
+{"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../../../src/types/stt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,MAAM,MAAM,SAAS,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;AAEjD,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,UAAU,EAAE,CAAC;CACrB;AAGD,MAAM,MAAM,oBAAoB,GAC5B,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,GACJ,IAAI,CAAC;AAET,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,EAAE,oBAAoB,CAAC;CACjC;AAED,MAAM,WAAW,uBAAuB;IACtC,cAAc,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,cAAc,CAAC;IAC9B,aAAa,EAAE,cAAc,CAAC;IAC9B,eAAe,EAAE,cAAc,CAAC;CACjC"}

package/lib/typescript/utils/SpeechToTextModule/ASR.d.ts

@@ -0,0 +1,27 @@
+import { DecodingOptions, Segment, SpeechToTextModelConfig, WordTuple } from '../../types/stt';
+export declare class ASR {
+    private nativeModule;
+    private tokenizerModule;
+    private timePrecision;
+    private maxDecodeLength;
+    private chunkSize;
+    private minChunkSamples;
+    private samplingRate;
+    private startOfTranscriptToken;
+    private endOfTextToken;
+    private timestampBeginToken;
+    load(model: SpeechToTextModelConfig, onDownloadProgressCallback: (progress: number) => void): Promise<void>;
+    private getInitialSequence;
+    private generate;
+    private softmaxWithTemperature;
+    private sampleFromDistribution;
+    private generateWithFallback;
+    private calculateWordLevelTimestamps;
+    private estimateWordTimestampsLinear;
+    transcribe(audio: number[], options: DecodingOptions): Promise<Segment[]>;
+    tsWords(segments: Segment[]): WordTuple[];
+    segmentsEndTs(res: Segment[]): number[];
+    encode(waveform: Float32Array): Promise<void>;
+    decode(tokens: number[]): Promise<Float32Array>;
+}
+//# sourceMappingURL=ASR.d.ts.map

package/lib/typescript/utils/SpeechToTextModule/ASR.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ASR.d.ts","sourceRoot":"","sources":["../../../../src/utils/SpeechToTextModule/ASR.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,eAAe,EACf,OAAO,EACP,uBAAuB,EAEvB,SAAS,EACV,MAAM,iBAAiB,CAAC;AAGzB,qBAAa,GAAG;IACd,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,eAAe,CAA0C;IAEjE,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,eAAe,CAAe;IACtC,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,eAAe,CAAqB;IAC5C,OAAO,CAAC,YAAY,CAAiB;IAErC,OAAO,CAAC,sBAAsB,CAAU;IACxC,OAAO,CAAC,cAAc,CAAU;IAChC,OAAO,CAAC,mBAAmB,CAAU;IAExB,IAAI,CACf,KAAK,EAAE,uBAAuB,EAC9B,0BAA0B,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI;YA8B1C,kBAAkB;YAgBlB,QAAQ;IAoCtB,OAAO,CAAC,sBAAsB;IAO9B,OAAO,CAAC,sBAAsB;YAYhB,oBAAoB;YA4BpB,4BAA4B;YA4D5B,4BAA4B;IAkC7B,UAAU,CACrB,KAAK,EAAE,MAAM,EAAE,EACf,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,OAAO,EAAE,CAAC;IA2Bd,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,EAAE;IAUzC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE;IAItB,MAAM,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7C,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;CAG7D"}

package/lib/typescript/utils/SpeechToTextModule/OnlineProcessor.d.ts

@@ -0,0 +1,23 @@
+import { DecodingOptions } from '../../types/stt';
+import { ASR } from './ASR';
+export declare class OnlineASRProcessor {
+    private asr;
+    private samplingRate;
+    audioBuffer: number[];
+    private transcriptBuffer;
+    private bufferTimeOffset;
+    private committed;
+    constructor(asr: ASR);
+    insertAudioChunk(audio: number[]): void;
+    processIter(options: DecodingOptions): Promise<{
+        committed: string;
+        nonCommitted: string;
+    }>;
+    private chunkCompletedSegment;
+    private chunkAt;
+    finish(): Promise<{
+        committed: string;
+    }>;
+    private toFlush;
+}
+//# sourceMappingURL=OnlineProcessor.d.ts.map

package/lib/typescript/utils/SpeechToTextModule/OnlineProcessor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OnlineProcessor.d.ts","sourceRoot":"","sources":["../../../../src/utils/SpeechToTextModule/OnlineProcessor.ts"],"names":[],"mappings":"AAEA,OAAO,EAAa,eAAe,EAAW,MAAM,iBAAiB,CAAC;AACtE,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAG5B,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,GAAG,CAAM;IAEjB,OAAO,CAAC,YAAY,CAAiB;IAC9B,WAAW,EAAE,MAAM,EAAE,CAAM;IAClC,OAAO,CAAC,gBAAgB,CAA4C;IACpE,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,SAAS,CAAmB;gBAExB,GAAG,EAAE,GAAG;IAIb,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE;IAI1B,WAAW,CAAC,OAAO,EAAE,eAAe;;;;IAoBjD,OAAO,CAAC,qBAAqB;IAqB7B,OAAO,CAAC,OAAO;IASF,MAAM;;;IAOnB,OAAO,CAAC,OAAO;CAMhB"}

package/lib/typescript/utils/SpeechToTextModule/hypothesisBuffer.d.ts

@@ -0,0 +1,13 @@
+import { WordTuple } from '../../types/stt';
+export declare class HypothesisBuffer {
+    private committedInBuffer;
+    private buffer;
+    private new;
+    private lastCommittedTime;
+    lastCommittedWord: string | null;
+    insert(newWords: WordTuple[], offset: number): void;
+    flush(): WordTuple[];
+    popCommitted(time: number): void;
+    complete(): WordTuple[];
+}
+//# sourceMappingURL=hypothesisBuffer.d.ts.map

package/lib/typescript/utils/SpeechToTextModule/hypothesisBuffer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hypothesisBuffer.d.ts","sourceRoot":"","sources":["../../../../src/utils/SpeechToTextModule/hypothesisBuffer.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,GAAG,CAAmB;IAE9B,OAAO,CAAC,iBAAiB,CAAa;IAC/B,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAQ;IAExC,MAAM,CAAC,QAAQ,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,MAAM;IAuC5C,KAAK,IAAI,SAAS,EAAE;IAkBpB,YAAY,CAAC,IAAI,EAAE,MAAM;IAMzB,QAAQ,IAAI,SAAS,EAAE;CAG/B"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-executorch",
-  "version": "0.5.1-rc.0",
+  "version": "0.5.1",
   "description": "An easy way to run AI models in React Native with ExecuTorch",
   "source": "./src/index.ts",
   "main": "./lib/module/index.js",
@@ -31,10 +31,12 @@
   ],
   "scripts": {
     "example": "yarn workspace react-native-executorch-example",
-    "typecheck": "tsc",
+    "typecheck": "tsc --noEmit",
     "lint": "eslint \"**/*.{js,ts,tsx}\"",
     "clean": "del-cli android/build example/android/build example/android/app/build example/ios/build lib",
-    "prepare": "bob build"
+    "prepare": "bob build",
+    "prepack": "cp ../../README.md ./README.md",
+    "postpack": "rm ./README.md"
   },
   "keywords": [
     "react-native",

package/src/constants/modelUrls.ts

@@ -1,10 +1,8 @@
 import { Platform } from 'react-native';
-import { AvailableModels } from '../types/stt';
 
 const URL_PREFIX =
   'https://huggingface.co/software-mansion/react-native-executorch';
-const VERSION_TAG = 'resolve/v0.4.0';
-const NEXT_VERSION_TAG = 'resolve/v0.5.0';
+const VERSION_TAG = 'resolve/v0.5.0';
 
 // LLMs
 
@@ -308,35 +306,70 @@ export const STYLE_TRANSFER_UDNIE = {
 };
 
 // S2T
-const MOONSHINE_TINY_DECODER_MODEL = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/xnnpack/moonshine_tiny_xnnpack_decoder.pte`;
-const MOONSHINE_TINY_ENCODER_MODEL = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/xnnpack/moonshine_tiny_xnnpack_encoder.pte`;
-const MOONSHINE_TOKENIZER = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/moonshine_tiny_tokenizer.json`;
-const WHISPER_TOKENIZER = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/whisper_tokenizer.json`;
-const WHISPER_TINY_DECODER_MODEL = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_xnnpack_decoder.pte`;
-const WHISPER_TINY_ENCODER_MODEL = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_xnnpack_encoder.pte`;
-const WHISPER_TINY_MULTILINGUAL_ENCODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/xnnpack_whisper_encoder.pte`;
-const WHISPER_TINY_MULTILINGUAL_DECODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/xnnpack_whisper_decoder.pte`;
-const WHISPER_TINY_MULTILINGUAL_TOKENIZER = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/tokenizer.json`;
-
-export const MOONSHINE_TINY = {
-  modelName: AvailableModels.MOONSHINE,
-  decoderSource: MOONSHINE_TINY_DECODER_MODEL,
-  encoderSource: MOONSHINE_TINY_ENCODER_MODEL,
-  tokenizerSource: MOONSHINE_TOKENIZER,
+const WHISPER_TINY_EN_TOKENIZER = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_TINY_EN_ENCODER = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_encoder_xnnpack.pte`;
+const WHISPER_TINY_EN_DECODER = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_decoder_xnnpack.pte`;
+
+const WHISPER_BASE_EN_TOKENIZER = `${URL_PREFIX}-whisper-base.en/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_BASE_EN_ENCODER = `${URL_PREFIX}-whisper-base.en/${VERSION_TAG}/xnnpack/whisper_base_en_encoder_xnnpack.pte`;
+const WHISPER_BASE_EN_DECODER = `${URL_PREFIX}-whisper-base.en/${VERSION_TAG}/xnnpack/whisper_base_en_decoder_xnnpack.pte`;
+
+const WHISPER_SMALL_EN_TOKENIZER = `${URL_PREFIX}-whisper-small.en/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_SMALL_EN_ENCODER = `${URL_PREFIX}-whisper-small.en/${VERSION_TAG}/xnnpack/whisper_small_en_encoder_xnnpack.pte`;
+const WHISPER_SMALL_EN_DECODER = `${URL_PREFIX}-whisper-small.en/${VERSION_TAG}/xnnpack/whisper_small_en_decoder_xnnpack.pte`;
+
+const WHISPER_TINY_TOKENIZER = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_TINY_ENCODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/whisper_tiny_encoder_xnnpack.pte`;
+const WHISPER_TINY_DECODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/whisper_tiny_decoder_xnnpack.pte`;
+
+const WHISPER_BASE_TOKENIZER = `${URL_PREFIX}-whisper-base/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_BASE_ENCODER_MODEL = `${URL_PREFIX}-whisper-base/${VERSION_TAG}/xnnpack/whisper_base_encoder_xnnpack.pte`;
+const WHISPER_BASE_DECODER_MODEL = `${URL_PREFIX}-whisper-base/${VERSION_TAG}/xnnpack/whisper_base_decoder_xnnpack.pte`;
+
+const WHISPER_SMALL_TOKENIZER = `${URL_PREFIX}-whisper-small/${VERSION_TAG}/tokenizer.json`;
+const WHISPER_SMALL_ENCODER_MODEL = `${URL_PREFIX}-whisper-small/${VERSION_TAG}/xnnpack/whisper_small_encoder_xnnpack.pte`;
+const WHISPER_SMALL_DECODER_MODEL = `${URL_PREFIX}-whisper-small/${VERSION_TAG}/xnnpack/whisper_small_decoder_xnnpack.pte`;
+
+export const WHISPER_TINY_EN = {
+  isMultilingual: false,
+  encoderSource: WHISPER_TINY_EN_ENCODER,
+  decoderSource: WHISPER_TINY_EN_DECODER,
+  tokenizerSource: WHISPER_TINY_EN_TOKENIZER,
+};
+
+export const WHISPER_BASE_EN = {
+  isMultilingual: false,
+  encoderSource: WHISPER_BASE_EN_ENCODER,
+  decoderSource: WHISPER_BASE_EN_DECODER,
+  tokenizerSource: WHISPER_BASE_EN_TOKENIZER,
+};
+
+export const WHISPER_SMALL_EN = {
+  isMultilingual: false,
+  encoderSource: WHISPER_SMALL_EN_ENCODER,
+  decoderSource: WHISPER_SMALL_EN_DECODER,
+  tokenizerSource: WHISPER_SMALL_EN_TOKENIZER,
 };
 
 export const WHISPER_TINY = {
-  modelName: AvailableModels.WHISPER,
-  decoderSource: WHISPER_TINY_DECODER_MODEL,
+  isMultilingual: true,
   encoderSource: WHISPER_TINY_ENCODER_MODEL,
-  tokenizerSource: WHISPER_TOKENIZER,
+  decoderSource: WHISPER_TINY_DECODER_MODEL,
+  tokenizerSource: WHISPER_TINY_TOKENIZER,
+};
+
+export const WHISPER_BASE = {
+  isMultilingual: true,
+  encoderSource: WHISPER_BASE_ENCODER_MODEL,
+  decoderSource: WHISPER_BASE_DECODER_MODEL,
+  tokenizerSource: WHISPER_BASE_TOKENIZER,
 };
 
-export const WHISPER_TINY_MULTILINGUAL = {
-  modelName: AvailableModels.WHISPER_MULTILINGUAL,
-  decoderSource: WHISPER_TINY_MULTILINGUAL_DECODER_MODEL,
-  encoderSource: WHISPER_TINY_MULTILINGUAL_ENCODER_MODEL,
-  tokenizerSource: WHISPER_TINY_MULTILINGUAL_TOKENIZER,
+export const WHISPER_SMALL = {
+  isMultilingual: true,
+  encoderSource: WHISPER_SMALL_ENCODER_MODEL,
+  decoderSource: WHISPER_SMALL_DECODER_MODEL,
+  tokenizerSource: WHISPER_SMALL_TOKENIZER,
 };
 
 // Image segmentation
@@ -347,23 +380,23 @@ export const DEEPLAB_V3_RESNET50 = {
 };
 
 // Image Embeddings
-const CLIP_VIT_BASE_PATCH32_IMAGE_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/clip-vit-base-patch32-vision_xnnpack.pte`;
+const CLIP_VIT_BASE_PATCH32_IMAGE_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${VERSION_TAG}/clip-vit-base-patch32-vision_xnnpack.pte`;
 
 export const CLIP_VIT_BASE_PATCH32_IMAGE = {
   modelSource: CLIP_VIT_BASE_PATCH32_IMAGE_MODEL,
 };
 
 // Text Embeddings
-const ALL_MINILM_L6_V2_MODEL = `${URL_PREFIX}-all-MiniLM-L6-v2/${NEXT_VERSION_TAG}/all-MiniLM-L6-v2_xnnpack.pte`;
-const ALL_MINILM_L6_V2_TOKENIZER = `${URL_PREFIX}-all-MiniLM-L6-v2/${NEXT_VERSION_TAG}/tokenizer.json`;
-const ALL_MPNET_BASE_V2_MODEL = `${URL_PREFIX}-all-mpnet-base-v2/${NEXT_VERSION_TAG}/all-mpnet-base-v2_xnnpack.pte`;
-const ALL_MPNET_BASE_V2_TOKENIZER = `${URL_PREFIX}-all-mpnet-base-v2/${NEXT_VERSION_TAG}/tokenizer.json`;
-const MULTI_QA_MINILM_L6_COS_V1_MODEL = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${NEXT_VERSION_TAG}/multi-qa-MiniLM-L6-cos-v1_xnnpack.pte`;
-const MULTI_QA_MINILM_L6_COS_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${NEXT_VERSION_TAG}/tokenizer.json`;
-const MULTI_QA_MPNET_BASE_DOT_V1_MODEL = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${NEXT_VERSION_TAG}/multi-qa-mpnet-base-dot-v1_xnnpack.pte`;
-const MULTI_QA_MPNET_BASE_DOT_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${NEXT_VERSION_TAG}/tokenizer.json`;
-const CLIP_VIT_BASE_PATCH32_TEXT_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/clip-vit-base-patch32-text_xnnpack.pte`;
-const CLIP_VIT_BASE_PATCH32_TEXT_TOKENIZER = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/tokenizer.json`;
+const ALL_MINILM_L6_V2_MODEL = `${URL_PREFIX}-all-MiniLM-L6-v2/${VERSION_TAG}/all-MiniLM-L6-v2_xnnpack.pte`;
+const ALL_MINILM_L6_V2_TOKENIZER = `${URL_PREFIX}-all-MiniLM-L6-v2/${VERSION_TAG}/tokenizer.json`;
+const ALL_MPNET_BASE_V2_MODEL = `${URL_PREFIX}-all-mpnet-base-v2/${VERSION_TAG}/all-mpnet-base-v2_xnnpack.pte`;
+const ALL_MPNET_BASE_V2_TOKENIZER = `${URL_PREFIX}-all-mpnet-base-v2/${VERSION_TAG}/tokenizer.json`;
+const MULTI_QA_MINILM_L6_COS_V1_MODEL = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${VERSION_TAG}/multi-qa-MiniLM-L6-cos-v1_xnnpack.pte`;
+const MULTI_QA_MINILM_L6_COS_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${VERSION_TAG}/tokenizer.json`;
+const MULTI_QA_MPNET_BASE_DOT_V1_MODEL = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${VERSION_TAG}/multi-qa-mpnet-base-dot-v1_xnnpack.pte`;
+const MULTI_QA_MPNET_BASE_DOT_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${VERSION_TAG}/tokenizer.json`;
+const CLIP_VIT_BASE_PATCH32_TEXT_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${VERSION_TAG}/clip-vit-base-patch32-text_xnnpack.pte`;
+const CLIP_VIT_BASE_PATCH32_TEXT_TOKENIZER = `${URL_PREFIX}-clip-vit-base-patch32/${VERSION_TAG}/tokenizer.json`;
 
 export const ALL_MINILM_L6_V2 = {
   modelSource: ALL_MINILM_L6_V2_MODEL,

package/src/constants/ocr/models.ts

@@ -2,7 +2,7 @@ import { alphabets, symbols } from './symbols';
 
 const URL_PREFIX =
   'https://huggingface.co/software-mansion/react-native-executorch';
-const VERSION_TAG = 'resolve/v0.4.0';
+const VERSION_TAG = 'resolve/v0.5.0';
 
 const DETECTOR_CRAFT_1280_MODEL = `${URL_PREFIX}-detector-craft/${VERSION_TAG}/xnnpack/xnnpack_craft_1280.pte`;
 const DETECTOR_CRAFT_800_MODEL = `${URL_PREFIX}-detector-craft/${VERSION_TAG}/xnnpack/xnnpack_craft_800.pte`;