Teaching a machine to feel ❤️
This UI depends on a trained tensorflowjs model that outputs 7 logits. ["Angry", "Disgust", "Fear", "Happy", "Sad", "Surprise", "Neutral"]
are the emotions we assume this network can feel.
We've setup a Colab notebook that trains a model and renders this UI. You can find it here.
You can modify this UI and run it locally with a converted model by running:
git clone https://github.com/vanpelt/emotion-detector
cd emotion-detector
npm install
npm run start