Update index.js

index.js

@@ -494,9 +494,9 @@ async function upscaleImage(images) {
   return new Promise(async (resolve, reject) => {
       try {
         const upscaler = new Upscaler();
-        const image = tfjs.node.decodeImage(images, 3);
+        const imageTensor = tf.tensor3d(imageBuffer, [imageBuffer.height, imageBuffer.width, 3]);
 
-        const tensor = await upscaler.upscale(image, {
+        const upscaledTensor = await upscaler.upscale(imageTensor, {
           scale: 4, // upscale by a factor of 4
           sharpen: true, // apply sharpening
           denoise: true, // apply denoising
@@ -504,15 +504,12 @@ async function upscaleImage(images) {
           colorEnhance: true, // enhance colors
         });
 
-        const upscaledTensor = await tfjs.node.encodePng(tensor)
-        //const typedArray = await upscaledTensor.data();
-        const arrayBuffer = Buffer.from(upscaledTensor);
+        const upscaledImageBuffer = await tf.tensor3dToBuffer(upscaledTensor);
 
         // dispose the tensors!
-        image.dispose();
-        tensor.dispose();;
-
-        resolve(arrayBuffer);
+        imageTensor.dispose();
+        upscaledTensor.dispose();
+        resolve(upscaledImageBuffer);
       } catch (error) {
         reject(error);
       }