This commit is contained in:
Dragon Fire
2024-03-23 21:30:08 -04:00
parent 00e90222a9
commit f744e5d536
+2 -2
View File
@@ -78,11 +78,11 @@ module.exports = class XiaoClient extends CommandClient {
} }
async loadNSFWModel() { async loadNSFWModel() {
const model = await nsfw.load( const nsfwModel = await nsfw.load(
`${url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'nsfw', 'web_model')).href}/`, `${url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'nsfw', 'web_model')).href}/`,
{ type: 'graph' } { type: 'graph' }
); );
this.nsfwModel = model; this.nsfwModel = nsfwModel;
return this.nsfwModel; return this.nsfwModel;
} }