Update config.json

config.json

@@ -1,35 +1,38 @@
 {
-  "url": "https://huggingface.co/FredZhang7/google-safesearch-mini-v2/resolve/main/pytorch_model.bin",
-  "num_classes": 3,
+  "classes": {
+    "0": "nsfw_gore",
+    "1": "nsfw_suggestive",
+    "2": "safe"
+  },
+  "classifier": "default",
+  "crop_pct": 0.875,
+  "first_conv": "conv2d",
+  "has_aux": false,
+  "input_channels": 3,
   "input_size": [
     3,
     299,
     299
   ],
-  "pool_size": [
-    8,
-    8
-  ],
-  "crop_pct": 0.8975,
   "interpolation": "bicubic",
+  "label_offset": 0,
   "mean": [
     0.5,
     0.5,
     0.5
   ],
+  "model_name": "safesearch_mini_v2",
+  "model_type": "safesearch_mini_v2",
+  "num_classes": 3,
+  "output_channels": 1536,
+  "pool_size": [
+    8,
+    8
+  ],
   "std": [
     0.5,
     0.5,
     0.5
   ],
-  "first_conv": "conv2d_1a.conv",
-  "classifier": "classif",
-  "label_offset": 1,
-  "num_features": 1536,
-  "architecture": "inception_resnet_v2",
-  "labels": [
-    "nsfw_gore",
-    "nsfw_suggestive",
-    "safe"
-  ]
+  "transformers_version": "4.26.1"
 }