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

---
 structures/Tensorflow.js | 1 +
 1 file changed, 1 insertion(+)

diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js
index 5c9a4e87..fe4fe878 100644
--- a/structures/Tensorflow.js
+++ b/structures/Tensorflow.js
@@ -83,6 +83,7 @@ module.exports = class Tensorflow {
 		const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]);
 		loadedImage.dispose();
 		stylePrediction.dispose();
+		stylizedImage.mul(255).toInt();
 		const buffer = await tf.node.encodeJpeg(stylizedImage.squeeze());
 		stylizedImage.dispose();
 		return Buffer.from(buffer);