Graph model?

This commit is contained in:
Dragon Fire
2021-03-22 17:53:04 -04:00
parent 8e5d76436a
commit ebafccd911
+2 -1
View File
@@ -188,7 +188,8 @@ module.exports = class XiaoClient extends CommandoClient {
async loadNSFWModel() {
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;
return this.nsfwModel;