Update config.json

config.json

@@ -1,47 +1,35 @@
 {
-  "architectures": [
-    "SafeSearchModel"
-  ],
-  "auto_map": {
-    "AutoConfig": "model.SafeSearchConfig",
-    "AutoModelForImageClassification": "model.SafeSearchModel"
-  },
-  "classes": {
-    "0": "nsfw_gore",
-    "1": "nsfw_suggestive",
-    "2": "safe"
-  },
-  "classifier": "default",
-  "crop_pct": 0.875,
-  "device": "cpu",
-  "first_conv": "conv2d_1a.conv",
-  "has_aux": false,
-  "input_channels": 3,
+  "url": "https://huggingface.co/FredZhang7/google-safesearch-mini-v2/resolve/main/pytorch_model.bin",
+  "num_classes": 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
   ],
-  "torch_dtype": "float32",
-  "transformers_version": "4.25.1"
-}
+  "first_conv": "conv2d_1a.conv",
+  "classifier": "classif",
+  "label_offset": 0,
+  "num_features": 1536,
+  "architecture": "inception_resnet_v2",
+  "labels": [
+    "nsfw_gore",
+    "nsfw_suggestive",
+    "safe"
+  ]
+}