diff --git a/README.md b/README.md index dc5fb18..4938c64 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,14 @@ # film-poster-genres -Predicting film genres from their posters with Tensorflow.js \ No newline at end of file +> Predicting film genres from their posters with Tensorflow.js + + +To use: + +```bash +node src/index.mjs train --input datasets/posters-256/ --output output/ +node src/index.mjs predict --input datasets/posters-256/validate/25410,Comedy,Family,Drama.jpg --ai-model output/checkpoints/49 2>/dev/null +``` + + +TODO: Fill out this README (and maybe move it to GitHub & make this repo a clone)