NSFW Classifier#

This does a few different things:

• Scaffolding for building a ViT classifier
  • Dataset creation
  • Model training
  • Wrapper for existing model, i.e. lets you call predict_single or predict_single__bytes
• Firehose consumer (via ATKafka) to scan images with the given model
• Ozone client to apply moderation labels based on results of the model