haiefff commited on
Commit
4b6d51e
1 Parent(s): 9eb71d6
Files changed (3) hide show
  1. config.json +2 -3
  2. onnx/model.onnx +3 -0
  3. preprocessor_config.json +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224",
3
  "_num_labels": 2,
4
  "architectures": [
5
  "ViTForImageClassification"
@@ -28,6 +28,5 @@
28
  "patch_size": 16,
29
  "problem_type": "single_label_classification",
30
  "qkv_bias": true,
31
- "torch_dtype": "float32",
32
- "transformers_version": "4.42.2"
33
  }
 
1
  {
2
+ "_name_or_path": "haiefff/nsfw-images-2",
3
  "_num_labels": 2,
4
  "architectures": [
5
  "ViTForImageClassification"
 
28
  "patch_size": 16,
29
  "problem_type": "single_label_classification",
30
  "qkv_bias": true,
31
+ "transformers_version": "4.33.2"
 
32
  }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1611d857495dcc8e87893df5e3709790aabd17756cfc257230d1112f54dfccee
3
+ size 343463896
preprocessor_config.json CHANGED
@@ -7,7 +7,7 @@
7
  0.5,
8
  0.5
9
  ],
10
- "image_processor_type": "ViTImageProcessor",
11
  "image_std": [
12
  0.5,
13
  0.5,
 
7
  0.5,
8
  0.5
9
  ],
10
+ "image_processor_type": "ViTFeatureExtractor",
11
  "image_std": [
12
  0.5,
13
  0.5,