arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-25 14:21:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

set nsfwjs model directly

Browse Source
This commit is contained in:
Dragon Fire
2024-05-13 23:30:44 -04:00
parent 22ee2a5497
commit 578f3e5155
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+1 -1
View File
@@ -16,7 +16,7 @@ module.exports = class Tensorflow {
} }
async loadNSFWJS() { async loadNSFWJS() {
const nsfwjs = await nsfw.load(); const nsfwjs = await nsfw.load('MobileNetV2');
this.nsfwjs = nsfwjs; this.nsfwjs = nsfwjs;
return this.nsfwjs; return this.nsfwjs;
} }