tfjs model loader

gistfile1.txt

  async loadModel(): Promise<io.ModelArtifacts> {
    return fetch(`${BUCKET_URL}/model.json`)
      .then((response: Response) => response.json())
      .then((modelJson) =>
        io.getModelArtifactsForJSON(modelJson, (weightsManifest: WeightsManifestConfig) =>
          fetch(`${BUCKET_URL}/group1-shard1of1.bin`)
            .then((response: Response) => response.arrayBuffer())
            .then((weightData: WeightData) => [io.getWeightSpecs(weightsManifest), weightData])
        )
      ) as Promise<io.ModelArtifacts>
  }

  async loadAssets(): Promise<Blob> {
    return fetch(`${BUCKET_URL}/assets.zip`).then((response: Response) => response.arrayBuffer()) as Promise<Blob>;
  }

  async load(): Promise<TFDFModel> {
    return await loadTFDFModel({
      loadModel: () => this.loadModel(),
      loadAssets: () => this.loadAssets(),
    });
  }