pixelspark
commented
2 years ago @haixuanTao this works pretty well now (tested on Chrome Canary with WebGPU enabled). API looks like this at the moment:
<script type="module">
import init, { Session, Input } from "/target/pkg/wonnx.js";
async function fetchBytes(url) {
const reply = await fetch(url);
const blob = await reply.arrayBuffer();
const arr = new Uint8Array(blob);
return arr;
}
async function run() {
try {
const [modelBytes, initResult] = await Promise.all([fetchBytes("models/single_relu.onnx"), init()])
const session = await Session.fromBytes(modelBytes);
const input = new Input();
input.insert("x", [13.0, -37.0]);
const result = await session.run(input);
console.log({result});
session.free();
}
catch(e) {
console.error(e, e.toString());
}
}
run();
</script>
haixuanTao
Still very much a work in progress, but a simple test works!