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

---
 Xiao.js | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Xiao.js b/Xiao.js
index 6ac21f77..9134be3f 100644
--- a/Xiao.js
+++ b/Xiao.js
@@ -266,6 +266,15 @@ client.on('ready', async () => {
 		client.logger.error(`[FACE DETECTOR] Failed to load face detector\n${err.stack}`);
 	}
 
+	// Set up stylize models
+	try {
+		await client.tensorflow.loadStyleModel();
+		await client.tensorflow.loadTransformerModel();
+		client.logger.info('[STYLIZE] Loaded stylize models.');
+	} catch (err) {
+		client.logger.error(`[STYLIZE] Failed to load stylize models\n${err.stack}`);
+	}
+
 	// Fetch all members
 	try {
 		for (const guild of client.guilds.cache.values()) {