更新API

This commit is contained in:
dengtongyu committed 2020-11-23 15:30:21 +08:00
1 parent 16769a8726
commit ece1e8065c
16 files changed
+161 -1

No files matched your search

+31
View File
@@ -0,0 +1,31 @@
'use strict';
// eslint-disable-next-line no-unused-vars
const nsfw = require('nsfwjs');
// eslint-disable-next-line no-unused-vars
const tf = require('@tensorflow/tfjs-node');
class AppBootHook {
constructor(app) {
this.app = app;
}
configWillLoad() {
// console.log('config: ', this.app.config.uploadHost);
}
async didLoad() {
console.log('nsfwModel 加载中···');
this.app.nsfwModel = await nsfw.load('file://./app/model/', { type: 'graph' });
// this.app.nsfwModel = await tf.loadGraphModel('file://./app/model/model.json');
console.log('nsfwModel 已经加州完毕!');
}
// async willReady() {
// }
// async didReady() {
// }
}
module.exports = AppBootHook;