diff --git a/packages/server/src/utils/indecent-prediction/index.js b/packages/server/src/utils/indecent-prediction/index.js
index 30fc87bd..a22ade17 100755
--- a/packages/server/src/utils/indecent-prediction/index.js
+++ b/packages/server/src/utils/indecent-prediction/index.js
@@ -1,4 +1,5 @@
 const tf = require("@tensorflow/tfjs-node")
+
 import * as nsfwjs from "nsfwjs/dist"
 import sharp from "sharp"
 
@@ -17,6 +18,10 @@ const imageToInput = (image, numChannels) => {
     return input
 }
 
+if (global.isProduction) {
+    tf.enableProdMode()
+}
+
 export default async (payload) => {
     let { url, image, channels = 3 } = payload