From 52186e74d01396de6ecfcb6761ae0a7433f4a529 Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Wed, 1 May 2024 23:17:18 -0400
Subject: [PATCH] Fix

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

diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js
index 8a852f2b..079e131c 100644
--- a/structures/Tensorflow.js
+++ b/structures/Tensorflow.js
@@ -1,6 +1,7 @@
 const tfnode = require('@tensorflow/tfjs-node');
 const nsfw = require('nsfwjs');
 const faceDetection = require('@tensorflow-models/face-detection');
+const faceModel = faceDetection.SupportedModels.MediaPipeFaceDetector;
 const url = require('url');
 const path = require('path');
 
@@ -22,7 +23,7 @@ module.exports = class Tensorflow {
 	}
 
 	async loadFaceDetector() {
-		const faceDetector = await faceDetection.createDetector(this.faceModel, { runtime: 'tfjs', maxFaces: 10 });
+		const faceDetector = await faceDetection.createDetector(faceModel, { runtime: 'tfjs', maxFaces: 10 });
 		this.faceDetector = faceDetector;
 		return this.faceDetector;
 	}