32 lines
839 B
JavaScript
32 lines
839 B
JavaScript
import { pipeline } from "@xenova/transformers";
|
|
import Jimp from "jimp";
|
|
|
|
async function main() {
|
|
const modelName = "Xenova/distilbert-base-uncased-finetuned-sst-2-english"; // Example model
|
|
|
|
// Load the model
|
|
const classifier = await pipeline("image-classification", modelName);
|
|
|
|
// Load the image
|
|
let image;
|
|
try {
|
|
image = await Jimp.read("/home/karan/Downloads/tomato.png");
|
|
image.rgba(true);
|
|
} catch (error) {
|
|
console.error("Error: Unable to open image. Check the file type or path.");
|
|
console.error(error);
|
|
return;
|
|
}
|
|
|
|
// Convert image to buffer
|
|
const buffer = await image.getBufferAsync(Jimp.MIME_PNG);
|
|
|
|
// Classify the image
|
|
const result = await classifier(buffer);
|
|
|
|
console.log("Predicted class:", result);
|
|
}
|
|
|
|
main().catch(console.error);
|
|
|