RageStudio/comty
1
0
Fork 0
mirror of https://github.com/ragestudio/comty.git synced 2025-06-10 02:54:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

enable tensorflow prod

Browse Source
This commit is contained in:
SrGooglo 2023-04-19 15:16:48 +00:00
parent 3438e3385c
commit f31aa86542
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/server/src/utils/indecent-prediction/index.js
View File

@ -1,4 +1,5 @@
const tf = require("@tensorflow/tfjs-node") const tf = require("@tensorflow/tfjs-node")
import * as nsfwjs from "nsfwjs/dist" import * as nsfwjs from "nsfwjs/dist"
import sharp from "sharp" import sharp from "sharp"
@ -17,6 +18,10 @@ const imageToInput = (image, numChannels) => {
return input return input
} }
if (global.isProduction) {
tf.enableProdMode()
}
export default async (payload) => { export default async (payload) => {
let { url, image, channels = 3 } = payload let { url, image, channels = 3 } = payload