diff --git a/mediapipe/tasks/web/core/fileset_resolver.ts b/mediapipe/tasks/web/core/fileset_resolver.ts index d4691243..9917035a 100644 --- a/mediapipe/tasks/web/core/fileset_resolver.ts +++ b/mediapipe/tasks/web/core/fileset_resolver.ts @@ -44,22 +44,14 @@ async function isSimdSupported(): Promise { } async function createFileset( - taskName: string, basePath: string = '.'): Promise { - if (await isSimdSupported()) { - return { - wasmLoaderPath: - `${basePath}/${taskName}_wasm_internal.js`, - wasmBinaryPath: - `${basePath}/${taskName}_wasm_internal.wasm`, - }; - } else { - return { - wasmLoaderPath: - `${basePath}/${taskName}_wasm_nosimd_internal.js`, - wasmBinaryPath: - `${basePath}/${taskName}_wasm_nosimd_internal.wasm`, - }; - } + taskName: string, basePath: string = ''): Promise { + const suffix = + await isSimdSupported() ? 'wasm_internal' : 'wasm_nosimd_internal'; + + return { + wasmLoaderPath: `${basePath}/${taskName}_${suffix}.js`, + wasmBinaryPath: `${basePath}/${taskName}_${suffix}.wasm`, + }; } // tslint:disable:class-as-namespace