This is the same demo written by Tensorflowjs only in my web and mobile editable format. Click on the black box and change anything and see if it works. If you like it, copy the code to your computer or Github. If it doesn't work refresh the page and start again

...