Merge pull request #2875 from entrylabs/issue/8906

issue/8906
entrylabs · Nov 26, 2024 · 37dd6bb · 37dd6bb
2 parents 91a40b1 + 879b92c
commit 37dd6bb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/src/class/AILearning.js b/src/class/AILearning.js
@@ -262,8 +262,8 @@ export default class AILearning {
         this.loadModel({ url, trainParam, tableData, isActive, classes });
     }
 
-    async reload() {
-        await this.#module?.reload?.();
+    async reload(url) {
+        await this.#module?.reload?.(url);
     }
 
     openInputPopup() {

diff --git a/src/class/learning/ImageLearning.js b/src/class/learning/ImageLearning.js
@@ -145,8 +145,8 @@ class ImageLearning {
         this.isLoaded = true;
     }
 
-    async reload() {
-        this.model = await tf.loadLayersModel(this.#url);
+    async reload(url) {
+        this.model = await tf.loadLayersModel(url || this.#url);
         this.isLoaded = true;
     }
 }

diff --git a/src/class/learning/LogisticRegression.js b/src/class/learning/LogisticRegression.js
@@ -77,8 +77,8 @@ class LogisticRegression extends LearningBase {
         model.dispose();
     }
 
-    async reload() {
-        this.model = await tf.loadLayersModel(this.url);
+    async reload(url) {
+        this.model = await tf.loadLayersModel(url || this.url);
         this.isLoaded = true;
     }
 

diff --git a/src/class/learning/SpeechClassification.js b/src/class/learning/SpeechClassification.js
@@ -72,8 +72,8 @@ class Classification {
         this.model = await tf.loadLayersModel(url);
     }
 
-    async reload() {
-        this.model = await tf.loadLayersModel(this.#url);
+    async reload(url) {
+        this.model = await tf.loadLayersModel(url || this.#url);
         this.isLoaded = true;
     }
 }