Merge pull request #4 from pivanov/master

[patch] - issue #3

Author: pivanov

index.js

@@ -30,6 +30,15 @@ const retryOperation = (operation, delay, times) => new Promise((resolve, reject
     });
 });
 
+const getHTMLCanvasElement = (imageUrl) => {
+  return new Promise((resolve, reject) => {
+    const image = new canvas.Image();
+    image.onload = () => resolve(image);
+    image.onerror = (e) => reject(e);
+    image.src = imageUrl;
+  });
+}
+
 const byProbabilty = (predictionA, predictionB) => {
   if (predictionA.probability > predictionB.probability) return -1;
   if (predictionA.probability < predictionB.probability) return 1;
@@ -83,9 +92,7 @@ class SashiDoTeachableMachine {
 
   async inference({ imageUrl }) {
     try {
-      const image = new canvas.Image();
-      image.src = imageUrl;
-
+      const image = await getHTMLCanvasElement(imageUrl);
       const predictions = await this.model.predict(image);
       return predictions.sort(byProbabilty);
     } catch (error) {

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sashido/teachablemachine-node",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Simplifies integration with Teachable Machine models from Google",
   "author": {
     "name": "Team SashiDo",