From be7ef7008712c8ce69475a53378b06dae27ae7a3 Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Wed, 8 May 2024 18:25:50 -0400
Subject: [PATCH] Fix

---
 structures/Tensorflow.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js
index 8e22f5b1..5c9a4e87 100644
--- a/structures/Tensorflow.js
+++ b/structures/Tensorflow.js
@@ -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);
 	}
 };