Removed test2.js, not req.
This commit is contained in:
@@ -1,31 +0,0 @@
|
||||
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);
|
||||
|
||||
Reference in New Issue
Block a user