This is a template website to host a TensorFlow JS (TFJS) version of image classification models online.
To host your own classification + X website:
- Copy this repository to your own repository
- Place the TFJS model JSON and shards to the
model_tfjs/
folder - Open
model_tfjs/model.json
and replace instances of "Functional" with "Model" - Overwrite the
CLASSES_NAMES
list inindex.js
- Make sure the image normalization steps in
index.js
is the same as training - Modify
index.html
as desired - Host on GitHub Pages
Learn more about exporting models from TensorFlow to TFJS here.
See more TFJS examples here.