Spaces:
Paused
Paused
Update index.js
Browse files
index.js
CHANGED
|
@@ -504,7 +504,7 @@ async function upscaleImage(images) {
|
|
| 504 |
colorEnhance: true, // enhance colors
|
| 505 |
});
|
| 506 |
|
| 507 |
-
const upscaledTensor = await
|
| 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!
|