arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-08 17:58:34 -04:00
parent e73d68c3e0
commit f9d8ab9883
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+5 -2
View File
@@ -3,6 +3,7 @@ const nsfw = require('nsfwjs');
const faceDetection = require('@tensorflow-models/face-detection');
const faceModel = faceDetection.SupportedModels.MediaPipeFaceDetector;
const path = require('path');
const url = require('url');
module.exports = class Tensorflow {
constructor(client) {
@@ -27,14 +28,16 @@ module.exports = class Tensorflow {
}
async loadStyleModel() {
const model = await tf.loadGraphModel(path.join(__dirname, '..', 'tf_models', 'style_js', 'model.json'));
const model = await tf.loadGraphModel(
url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'style_js', 'model.json')).href
);
this.styleModel = model;
return this.styleModel;
}
async loadTransformerModel() {
const model = await tf.loadGraphModel(
path.join(__dirname, '..', 'tf_models', 'transformer_separable_js', 'model.json')
url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'transformer_separable_js', 'model.json')).href
);
this.transformerModel = model;
return this.transformerModel;