Update model_weights.js

model_weights.js

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