Merge pull request #1247 from danidask/main

joeyklee · web-flow · commit 22112fe5b3c3 · 2022-02-07T20:24:24.000-05:00
perf(ImageClassifier): use urls directly from teachablemachine
diff --git a/src/ImageClassifier/index.js b/src/ImageClassifier/index.js
@@ -65,7 +65,10 @@ class ImageClassifier {
             this.modelToUse = null;
         }
       } else {
+        // its a url, we expect to find model.json
         this.modelUrl = modelNameOrUrl;
+        // The teachablemachine urls end with a slash, so add model.json to complete the full path
+        if (this.modelUrl.endsWith('/')) this.modelUrl += "model.json";
       }
     }
     // Load the model

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,10 @@ class ImageClassifier {`
`65`	`65`	`this.modelToUse = null;`
`66`	`66`	`}`
`67`	`67`	`} else {`
	`68`	`+ // its a url, we expect to find model.json`
`68`	`69`	`this.modelUrl = modelNameOrUrl;`
	`70`	`+ // The teachablemachine urls end with a slash, so add model.json to complete the full path`
	`71`	`+ if (this.modelUrl.endsWith('/')) this.modelUrl += "model.json";`
`69`	`72`	`}`
`70`	`73`	`}`
`71`	`74`	`// Load the model`