Skip to content

leonardblier/neural-models

Repository files navigation

To use the porn detection service, you need the model file porn.h5 (e.g., /mnt/data/NeuralModels/porn.h5 in the code snippet).

docker build -f Dockerfile.dependencies -t neural-models-dependencies .
docker build --no-cache -t neural-models .
docker run --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm -p 5234:5234 -v /mnt/data/NeuralModels:/data neural-models
curl -XPOST -F "file1=@my_porn_image.jpg" -F "file2=@my_non_porn_image.jpg" http://0.0.0.0:5234/detect
# {"results": [{"id": "file1", "prob": 1.0}, {"id": "file2", "prob": 0.0}]}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages