const catImage = document.getElementById('cat');
const math = new dl.NDArrayMathGPU();
const squeezeNet = new squeezenet.SqueezeNet(math);

await squeezeNet.load();
const image = dl.Array3D.fromPixels(catImage);
const inferenceResult = await squeezeNet.predict(image);

const topClassesToProbs = await squeezeNet.getTopKClasses(inferenceResult.logits, 10);
document.getElementById('myDivOut').innerHTML = ''   // clear the div
for (const className in topClassesToProbs) { 
   document.getElementById('myDivOut').innerHTML += topClassesToProbs[className].toFixed(5) + ' : '+ className + '<br>'
}