Khrisna commited on
Commit
b58aa14
·
verified ·
1 Parent(s): 5035a76

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +1 -1
index.js CHANGED
@@ -504,7 +504,7 @@ async function upscaleImage(images) {
504
  colorEnhance: true, // enhance colors
505
  });
506
 
507
- const upscaledTensor = await tf.node.encodePng(tensor);
508
  const arrayBuffer = await upscaledTensor.arrayBuffer();
509
 
510
  // dispose the tensors!
 
504
  colorEnhance: true, // enhance colors
505
  });
506
 
507
+ const upscaledTensor = await tfjs.node.encodePng(tensor);
508
  const arrayBuffer = await upscaledTensor.arrayBuffer();
509
 
510
  // dispose the tensors!