How to use machine learning in your JS applications

Victor Muštar — Mon, 28 Feb 2022 15:38:07 +0000

Maybe, like me, you always wanted to use machine learning in your JS applications: you had some problems you think ML could solve (like summarization, sentiment analysis, object detection in images, etc, etc...) or you just wanted to have fun.

But the need to learn python, Machine Learning, and the whole ML stack is pulling you away. Building a good front-end is already hard enough...

So here is a super easy way to use machine learning in your apps (through the Hugging Face inference API, note that the free-tier is quite generous).

Start by choosing a model here: hf.co/models (+30k available models, a lot of them state-of-the-art). If you don't know which one to choose you can start by checking the different tasks available.

Once you found a model that you like just click Deploy -> Inference API then select JavaScript. For example, here is the one I usually use to do summarization: https://huggingface.co/sshleifer/distilbart-cnn-6-6

And that's it! You can now use this super powerful ML model anywhere you want, or even try it directly in your JS console by pasting the JS snippet (here is an example for summarization):

Good luck and have fun!
Victor

DEV Community: Victor Muštar

How to use machine learning in your JS applications