Integrate stable diffusion safety checker / NSFW image detection into web service like FastAPI.
Project description
Safety Checker
Safety checker is a model used in stable diffusion pipeline, aimed to identify NSFW image. Here is the official description.
This project extracts its safety checker into a independent function to provide a conventional way to detect NSFW image in deep neural network.
See test_imgs.py to get start.
TODO
- add standalone safety checker implementation that depends on pytorch only. provide strength config (if possible) for NSFW detection
- add FastAPI integration
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
safety_checker-0.3.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for safety_checker-0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bd3252f15af587bc12333593e993c8db3c75e37fdbcb2487d3754a45be1a40f1 |
|
MD5 | 5e7de3d7301712bbeaafaa6d7d35c5aa |
|
BLAKE2b-256 | ff6b502d05359eaa61873fae2e723d50753eb64ab2b53e5b3bcf008af2fb1b05 |