search

notAI-tech / NudeNet

Lightweight nudity detection
https://nudenet.notai.tech/
GNU Affero General Public License v3.0
1.76k stars 342 forks source link

How to deploy a nudenet detector on AWS. #70

Open UtkarshBhardwaj007 opened 3 years ago

UtkarshBhardwaj007 commented 3 years ago

I've been struggling with deploying this as an endpoint on AWS. Any help would be appreciated.

bedapudi6788 commented 3 years ago

Just run sudo docker run -d -p8080:8080 notaitech/nudenet:classifier on any ec2 machine. It will start the classifier on CPU as an http api.

https://github.com/notAI-tech/fastDeploy/blob/master/cli/fastDeploy-file_client.py shows an example on how to send a image in a post request and get the prediction.

UtkarshBhardwaj007 commented 3 years ago

Thanks for the quick reply. I was actually trying to do it from within Sagemaker by making a Tensorflow model using the checkpoints available as a tar file and then calling the "Deploy" function. I am just running into a swarm of errors while doing this. Could you assist in this?

bedapudi6788 commented 3 years ago

sorry, I don't have time at the moment to dig into this.