mirror of
https://github.com/arthur-pbty/xiao.git
synced 2026-06-03 23:36:43 +02:00
set nsfwjs model directly
This commit is contained in:
@@ -16,7 +16,7 @@ module.exports = class Tensorflow {
|
||||
}
|
||||
|
||||
async loadNSFWJS() {
|
||||
const nsfwjs = await nsfw.load();
|
||||
const nsfwjs = await nsfw.load('MobileNetV2');
|
||||
this.nsfwjs = nsfwjs;
|
||||
return this.nsfwjs;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user