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
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.5.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for safety_checker-0.5-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 251151db480c3da8371f4a1e327884170d770967988a5a9aa773f15465755263 |
|
| MD5 | 22cad45c1f6108c5b9502150f5afe6d8 |
|
| BLAKE2b-256 | 9575ee145e8cbeaedec59f0d72a6b20dc8fbcb561414584985c3473f08ac1219 |
