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

Graph model?

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 17:53:04 -04:00
parent 8e5d76436a
commit ebafccd911
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Client.js
+2 -1
View File
@@ -188,7 +188,8 @@ module.exports = class XiaoClient extends CommandoClient {
async loadNSFWModel() { async loadNSFWModel() {
const model = await nsfw.load( const model = await nsfw.load(
url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'nsfw', 'web_model', 'model.json')) url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'nsfw', 'web_model', 'model.json')),
{ type: 'graph' }
); );
this.nsfwModel = model; this.nsfwModel = model;
return this.nsfwModel; return this.nsfwModel;