Load Tensorflow JS model from Google Drive

tfjs-drive-deploy-load-model.js

async function loadModel() {
    google.script.run.withSuccessHandler(async modelFiles => {
        const modelJson = new File([modelFiles[0]], "model.json", { type: "application/json" })
        const modelWeights = new File([Uint8Array.from(modelFiles[1])], "group1-shard1of1.bin")
        model = await tf.loadLayersModel(tf.io.browserFiles([modelJson, modelWeights]))
        model.summary();
    }).withFailureHandler(onFailure).loadModelFromDrive();
}