kikomiko's picture
init based on submission
5a74c0d
raw
history blame contribute delete
660 Bytes
// https://nextjs.org/docs/app/building-your-application/routing/route-handlers
import { pipeline } from "@huggingface/transformers";
// NOTE: We attach the classifier to the global object to avoid unnecessary reloads during development
const classifier = (globalThis.classifier ??= await pipeline(
"text-classification",
"Xenova/distilbert-base-uncased-finetuned-sst-2-english",
));
export async function GET(request) {
const text = request.nextUrl.searchParams.get("text");
if (!text) {
return Response.json({ message: "No text provided" }, { status: 400 });
}
const result = await classifier(text);
return Response.json(result[0]);
}