Update index.js
Browse files
index.js
CHANGED
@@ -494,7 +494,7 @@ async function upscaleImage(imageBuffer) {
|
|
494 |
return new Promise(async (resolve, reject) => {
|
495 |
try {
|
496 |
const upscaler = new Upscaler();
|
497 |
-
const imageTensor = tfjs.tensor3d(new
|
498 |
|
499 |
const upscaledTensor = await upscaler.upscale(imageTensor, {
|
500 |
scale: 4, // upscale by a factor of 4
|
|
|
494 |
return new Promise(async (resolve, reject) => {
|
495 |
try {
|
496 |
const upscaler = new Upscaler();
|
497 |
+
const imageTensor = tfjs.tensor3d(new Uint8Array(imageBuffer), [imageBuffer.height, imageBuffer.width, 3]);
|
498 |
|
499 |
const upscaledTensor = await upscaler.upscale(imageTensor, {
|
500 |
scale: 4, // upscale by a factor of 4
|