fixed issue #14, release 1.1.3

Author: bonastreyair

.gitignore

@@ -1,2 +1,3 @@
 node_modules
-*.DS*
+*.DS*
+.vscode

CHANGELOG.md

@@ -1,6 +1,13 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+## [1.1.3] - 2020-06-05
+### Fixed
+ - Prediction does not work when save_image's box is checked - [#14](https://github.com/bonastreyair/node-red-contrib-teachable-machine/issues/14)
+
+### Changed
+ - Dependancy is now `@tensorflow/tfjs v1.3.1` instead of `@tensorflow/tfjs-node v1.4.0`, to match teachable machine correct dependencies
+
 ## [1.1.2] - 2020-05-03
 ### Added
  - PayPal donation badge in `README` file

package.json

@@ -1,9 +1,9 @@
 {
   "name": "node-red-contrib-teachable-machine",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "description": "Simplifies integration with Teachable Machine models from Google",
   "dependencies": {
-    "@tensorflow/tfjs-node": "1.4.0",
+    "@tensorflow/tfjs": "1.3.1",
     "@teachablemachine/image": "0.8.4",
     "jsdom": "16.2.2",
     "canvas": "2.6.1"

teachable_machine.js

@@ -9,7 +9,7 @@ module.exports = function (RED) {
   const canvas = require('canvas')
   global.fetch = require('node-fetch')
   // Teachable Machine needs global scope of HTMLVideoElement class to do a check
-  global.HTMLVideoElement = class HTMLVideoElement {}
+  global.HTMLVideoElement = class HTMLVideoElement { }
 
   function setNodeStatus (node, status) {
     switch (status) {
@@ -147,8 +147,9 @@ module.exports = function (RED) {
     node.on('input', function (msg) {
       try {
         if (node.ready && node.modelUrl !== '') {
-          if (node.passThrough) { msg.image = msg.payload }
+          msg.image = msg.payload
           inference(msg)
+          if (!node.passThrough) { delete msg.image }
         } else {
           node.error('model is not ready')
         }