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

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

diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js
index 37c72ea1..8e22f5b1 100644
--- a/structures/Tensorflow.js
+++ b/structures/Tensorflow.js
@@ -80,7 +80,7 @@ module.exports = class Tensorflow {
 		styleTensor.dispose();
 		const stylePrediction = await this.styleModel.predict(loadedStyle);
 		loadedStyle.dispose();
-		const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction.squeeze()]);
+		const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]);
 		loadedImage.dispose();
 		stylePrediction.dispose();
 		const buffer = await tf.node.encodePng(stylizedImage.squeeze());