web-ai-toolkit
Version:
AI powered features on the web made easy
25 lines (24 loc) • 678 B
JavaScript
import { _ as n, a as o } from "./transformers-kV3s9sef.js";
import { w as c } from "./utils-DJ16RMae.js";
let s;
async function p(e, a = "Xenova/resnet-50") {
return new Promise(async (r, t) => {
try {
s || await w(a), typeof e != "string" && (e = URL.createObjectURL(e));
const i = await s(e);
r(i);
} catch (i) {
t(i);
}
});
}
async function w(e) {
return new Promise(async (a) => {
s || (n.allowLocalModels = !1, n.useBrowserCache = !1, s = await o("image-classification", e || "Xenova/resnet-50", {
device: navigator.ml ? "webnn-npu" : await c() ? "webgpu" : "wasm"
})), a();
});
}
export {
p as runClassifier
};