From 5cefbba75777547f97abd92affcf9ef10ac36163 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radam=C3=A9s=20Ajna?= Date: Thu, 14 Sep 2023 22:30:50 -0700 Subject: [PATCH] minor UI fixes (#856) * fixes * remove listener * remove event listener --- candle-wasm-examples/llama2-c/lib-example.html | 12 +++++++++--- candle-wasm-examples/whisper/lib-example.html | 9 ++++++--- candle-wasm-examples/yolo/lib-example.html | 17 ++++++++++++----- 3 files changed, 27 insertions(+), 11 deletions(-) diff --git a/candle-wasm-examples/llama2-c/lib-example.html b/candle-wasm-examples/llama2-c/lib-example.html index 22b12517..86fe9811 100644 --- a/candle-wasm-examples/llama2-c/lib-example.html +++ b/candle-wasm-examples/llama2-c/lib-example.html @@ -6,7 +6,7 @@ - + @@ -113,8 +113,14 @@ const handleMessage = (event) => { const { status, error, message, prompt, sentence } = event.data; if (status) updateStatus(event.data); - if (error) reject(new Error(error)); - if (status === "complete") resolve(event.data); + if (error) { + llamaWorker.removeEventListener("message", handleMessage); + reject(new Error(error)); + } + if (status === "complete") { + llamaWorker.removeEventListener("message", handleMessage); + resolve(event.data); + } }; controller.signal.addEventListener("abort", handleAbort); diff --git a/candle-wasm-examples/whisper/lib-example.html b/candle-wasm-examples/whisper/lib-example.html index ad48072b..3cfd87a7 100644 --- a/candle-wasm-examples/whisper/lib-example.html +++ b/candle-wasm-examples/whisper/lib-example.html @@ -6,7 +6,7 @@ - + @@ -51,18 +51,21 @@ mel_filtersURL, audioURL, }); - whisperWorker.addEventListener("message", (event) => { + function messageHandler(event) { console.log(event.data); if ("status" in event.data) { updateStatus(event.data); } if ("error" in event.data) { + whisperWorker.removeEventListener("message", messageHandler); reject(new Error(event.data.error)); } if (event.data.status === "complete") { + whisperWorker.removeEventListener("message", messageHandler); resolve(event.data); } - }); + } + whisperWorker.addEventListener("message", messageHandler); }); } diff --git a/candle-wasm-examples/yolo/lib-example.html b/candle-wasm-examples/yolo/lib-example.html index 8f7d07c7..d9f18975 100644 --- a/candle-wasm-examples/yolo/lib-example.html +++ b/candle-wasm-examples/yolo/lib-example.html @@ -6,7 +6,7 @@ - + @@ -199,17 +199,21 @@ confidence, iou_threshold, }); - yoloWorker.addEventListener("message", (event) => { + function handleMessage(event) { + console.log("message", event.data); if ("status" in event.data) { updateStatus(event.data.status); } if ("error" in event.data) { + yoloWorker.removeEventListener("message", handleMessage); reject(new Error(event.data.error)); } if (event.data.status === "complete") { + yoloWorker.removeEventListener("message", handleMessage); resolve(event.data); } - }); + } + yoloWorker.addEventListener("message", handleMessage); }); } // add event listener to detect button @@ -393,7 +397,7 @@ -
+
-
+

Examples: