YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
const tf = require('@tensorflow/tfjs');
const tfnode = require('@tensorflow/tfjs-node');
async function loadModel(){
const handler = tfnode.io.fileSystem('checkpoint/model.json');
const model = await tf.loadGraphModel(handler);
console.log("Model loaded")
const zeros = tf.randomNormal([8, 3, 224, 224]);
const out = model.predict(zeros);
console.log(out);
out.print();
}
loadModel();