Spaces:

Topcoder-Platform
/

space-demo-template

Runtime error

space-demo-template / app /api /classify /route.js

init based on submission

5a74c0d over 1 year ago

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]);
}

	// 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]);
	}