Models
private-detector.onnx
An image classifier that detect lewd images: https://github.com/bumble-tech/private-detector:
- Download the saved model here: https://storage.googleapis.com/private_detector/private_detector_with_frozen.zip
- Install tensorflow-onnx
- Run
python -m tf2onnx.convert --saved-model model/saved_model --output private-detector.onnx
toxic-comment
This model is a fine-tuned version of the DistilBERT model to classify toxic comments: https://huggingface.co/martin-ha/toxic-comment-model
- Run
optimum-cli export onnx --model martin-ha/toxic-comment-model toxic-comment/
