fix: onnx model loading

2024-01-03 12:14:48 +08:00
parent a9c3075b70
commit 7d0cfb6e5b
1 changed files with 4 additions and 2 deletions
@@ -6,11 +6,13 @@ import { modelData, onnxMaskToImage } from "./onnx_helper.js";
 ort.env.wasm.wasmPaths = "https://cdn.jsdelivr.net/npm/onnxruntime-web/dist/";

 export let model = null;
+let modelType = null;

 // Initialize the ONNX model
-export const initModel = async (modelType) => {
+export const initModel = async (type) => {
  try {
-    if (!model) {
+    if (!model || modelType !== type) {
+      modelType = type;
      model = await ort.InferenceSession.create(
        `${location.protocol}//${location.host}/sam_model?type=${modelType}`
      );