Update model_weights.js

model_weights.js

@@ -43,6 +43,8 @@ oReq.onload = function (oEvent) {
   }
 };
 
+tf.setBackend('wasm');
+
 tf.ready().then( function() {
         tf.enableProdMode();
         oReq.send(null);