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.4.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for safety_checker-0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ddbe4dc06c407e408366545757b820bbf681d7fa4f9e0575cc5ca2174667d8bd |
|
MD5 | 22ce64924ab9e4ea447e1633613e9924 |
|
BLAKE2b-256 | 020ee96269f2c5ab0553e3415c4ce1ce6c025ef1e402f445eb60c7ccfed02ac3 |