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.
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.2.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for safety_checker-0.2-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 92ca8b64d0daae378a8b6d0249c7ecb3ce7eeaee96e9882d85a153728c4eadd4 |
|
| MD5 | f654a19ad3069a9996a97ad9f36e7fa6 |
|
| BLAKE2b-256 | 7ff03c7caec6d085613056662370d85905d1a6d0f1d6c57fc90daa639e885596 |
