config.json

{
  "architectures": [
    "tf_efficientnet_b0.ns_jft_in1k"
  ],
  "model_type": "image-classification",
  "library_name": "onnx",
  "task": "image-classification",
  "labels": [
    "shit",
    "not_shit"
  ],
  "positive_label": "shit",
  "positive_threshold": 0.149,
  "preprocessor": {
    "input_size": 256,
    "mean": [
      0.485,
      0.456,
      0.406
    ],
    "std": [
      0.229,
      0.224,
      0.225
    ],
    "resize": "shortest_edge_then_center_crop"
  }
}