mrlectus's picture
Upload 7 files
01e2221 verified
raw
history blame contribute delete
567 Bytes
import { pipeline, env } from "@huggingface/transformers";
import { NextRequest, NextResponse } from "next/server";
env.allowRemoteModels = true;
env.allowLocalModels=false;
env.cacheDir = "./.cache";
export const POST = async (request: NextRequest) => {
try{
const { text, code } = await request.json();
const translator = await pipeline("translation", "Xenova/m2m100_418M");
const out = await translator(text, {
src_lang: "en",
tgt_lang: code,
} as never);
return NextResponse.json(out);
}catch(e){
console.error(e)
}
};