jaroslaw-weber / hotdog-not-hotdog

Pytorch model recognizing hotdogs and not-hotdogs deployed on flask. Inspired by Sillicon Valley TV series.

Hotdog or not hotdog?

https://www.youtube.com/results?search_query=sillicon+valley+hotdog Pytorch model recognizing hotdogs and not-hotdogs deployed on flask. Inspired by Sillicon Valley TV series.

How to use:

1. Put training data inside train_set folder
2. Finetune model by running finetunning_model.py
3. Run not_hotdog.py to start a flask server
4. Open a browser and go to http://localhost:5000/ and upload picture