cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not use full filename when FileLocator decides which asset to load

Browse Source 
Fixes https://github.com/google/mediapipe/issues/4819

PiperOrigin-RevId: 569506907
This commit is contained in:
Sebastian Schmidt 2023-09-29 09:05:49 -07:00 committed by Copybara-Service
parent 09a51bcdeb
commit d4561fb5c2
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mediapipe/tasks/web/core/task_runner.ts
View File

@ -52,13 +52,14 @@ export async function createTaskRunner<T extends TaskRunner>(
fileset: WasmFileset, options: TaskRunnerOptions): Promise<T> {
const fileLocator: FileLocator = {
locateFile(file): string {
const wasm = fileset.wasmBinaryPath.toString();
if (wasm.includes(file)) {
return wasm;
}
const asset = fileset.assetBinaryPath?.toString();
if (asset?.includes(file)) {
return asset;
// We currently only use a single .wasm file and a single .data file (for
// the tasks that have to load assets). We need to revisit how we
// initialize the file locator if we ever need to differentiate between
// diffferent files.
if (file.endsWith('.wasm')) {
return fileset.wasmBinaryPath.toString();
} else if (fileset.assetBinaryPath && file.endsWith('.data')) {
return fileset.assetBinaryPath.toString();
}
return file;
}