A Voila web-app to classify the user-provided image into one of these classes:
- anime
- Disney animation
- soviet animation
The app is powered by fastai / PyTorch.
Click this link to render it as a web-app:
One can easily modify the code to make it work with other classes too. See the Jupyter notebook for details. In short, you just need to:
- replace the classes names with your own (e.g. "cats", "dogs")
- replace the Bing image search API key (you can get a 7-days demo for free)
- uncomment a few lines of code:
- the get_new_model
- the lines about additional dependencies, as indicated in the notebeook
- run all cells in the notebook to generate the new trained model
