arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-08 18:25:50 -04:00
parent f40ded5a85
commit be7ef70087
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+2 -2
View File
@@ -83,8 +83,8 @@ module.exports = class Tensorflow {
const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]);
loadedImage.dispose();
stylePrediction.dispose();
const buffer = await tf.node.encodePng(stylizedImage.squeeze());
const buffer = await tf.node.encodeJpeg(stylizedImage.squeeze());
stylizedImage.dispose();
return buffer;
return Buffer.from(buffer);
}
};