rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-01-19 04:50:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1454 from valadaptive/classify-router

Browse Source 
Use Express router for classify endpoint
This commit is contained in:
Cohee 2023-12-04 21:42:40 +02:00 committed by GitHub
commit 88396935cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 43 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server.js
View File

@ -3621,7 +3621,7 @@ require('./src/endpoints/vectors').registerEndpoints(app, jsonParser);
require('./src/endpoints/translate').registerEndpoints(app, jsonParser); require('./src/endpoints/translate').registerEndpoints(app, jsonParser);
// Emotion classification // Emotion classification
require('./src/endpoints/classify').registerEndpoints(app, jsonParser); app.use('/api/extra/classify', require('./src/endpoints/classify').router);
// Image captioning // Image captioning
app.use('/api/extra/caption', require('./src/endpoints/caption').router); app.use('/api/extra/caption', require('./src/endpoints/caption').router);

25
src/endpoints/classify.js
View File

@ -1,13 +1,13 @@
const express = require('express');
const { jsonParser } = require('../express-common');
const TASK = 'text-classification'; const TASK = 'text-classification';
/** const router = express.Router();
* @param {import("express").Express} app
* @param {any} jsonParser
*/
function registerEndpoints(app, jsonParser) {
const cacheObject = {};
app.post('/api/extra/classify/labels', jsonParser, async (req, res) => { const cacheObject = {};
router.post('/labels', jsonParser, async (req, res) => {
try { try {
const module = await import('../transformers.mjs'); const module = await import('../transformers.mjs');
const pipe = await module.default.getPipeline(TASK); const pipe = await module.default.getPipeline(TASK);
@ -17,9 +17,9 @@ function registerEndpoints(app, jsonParser) {
console.error(error); console.error(error);
return res.sendStatus(500); return res.sendStatus(500);
} }
}); });
app.post('/api/extra/classify', jsonParser, async (req, res) => { router.post('/', jsonParser, async (req, res) => {
try { try {
const { text } = req.body; const { text } = req.body;
@ -45,9 +45,6 @@ function registerEndpoints(app, jsonParser) {
console.error(error); console.error(error);
return res.sendStatus(500); return res.sendStatus(500);
} }
}); });
}
module.exports = { module.exports = { router };
registerEndpoints,
};