/emotion-detector

Emotion detector with tfjs

Primary LanguageJavaScript

Emotion Detector

Teaching a machine to feel ❤️

How it works

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.

Quickstart

We've setup a Colab notebook that trains a model and renders this UI. You can find it here.

Development

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