ReVanced/revanced-bots
1
0
Fork 0
mirror of https://github.com/ReVanced/revanced-bots.git synced 2026-01-26 12:41:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: addTrainData not being added to the events

Browse Source
This commit is contained in:
GramingFoxTeam
2022-11-15 19:57:54 +03:00
parent 489ab247a6
commit 817fa92b58
3 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
server/events/addTrainData.js
View File

@@ -2,7 +2,7 @@ import { readFileSync, writeFileSync } from 'node:fs';
import { join } from 'node:path';
export default function addTrainData(eventData) {
const file = readFileSync(join(global.__dirname, global.config.fasttext.trainData), 'utf-8');
const file = readFileSync(join(global.__dirname, global.config.fasttext.trainFile), 'utf-8');
const data = file.split('\n');
const { label, text } = eventData;
@@ -10,7 +10,7 @@ export default function addTrainData(eventData) {
data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);
writeFileSync(join(global.__dirname, global.config.fasttext.trainData), data.join('\n'));
writeFileSync(join(global.__dirname, global.config.fasttext.trainFile), data.join('\n'));
return;
}

4
server/events/index.js
View File

@@ -1,9 +1,11 @@
import runAI from './ai.js';
import trainAI from './trainAI.js';
import runOCR from './ocr.js';
import addTrainData from './addTrainData.js';
export {
runAI,
trainAI,
runOCR
runOCR,
addTrainData
}