search

GantMan / nsfw_model

Keras model of NSFW detector
Other
1.76k stars 273 forks source link

How do I use the model in TensorFlow JS? #65

Closed pottekkat closed 1 year ago

pottekkat commented 4 years ago

Can anyone share the code of how I can use this model in tfjs?

const model = await tf.loadLayersModel('models/model.json')

I downloaded the model zip and loaded it as follows. How can I then use this in javascript code to access the model and classify images

GantMan commented 4 years ago

Best way to use this in TFJS is to use NSFWJS which is wrapped around this https://github.com/infinitered/nsfwjs

It even has GIF support

pottekkat commented 4 years ago

Thanks I will check it out!