commit files to HF hub

.gitattributes

@@ -36,3 +36,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 images/female.jpg filter=lfs diff=lfs merge=lfs -text
 images/male.jpg filter=lfs diff=lfs merge=lfs -text
 images/child.jpg filter=lfs diff=lfs merge=lfs -text
+images/adult_female.jpg filter=lfs diff=lfs merge=lfs -text
+images/adult_male.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -15,7 +15,7 @@ model-index:
     metrics:
       - name: Accuracy
         type: accuracy
-        value: 0.7611940503120422
+        value: 0.8507462739944458
 ---
 
 # sgdkn/gender-classification-hp
@@ -31,14 +31,14 @@ Report any issues with the demo at the [github repo](https://github.com/nateraw/
 ## Example Images
 
 
-#### child
+#### adult female
 
-![child](images/child.jpg)
+![adult female](images/adult_female.jpg)
 
-#### female
+#### adult male
 
-![female](images/female.jpg)
+![adult male](images/adult_male.jpg)
 
-#### male
+#### child
 
-![male](images/male.jpg)
+![child](images/child.jpg)

config.json

@@ -9,17 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "child",
-    "1": "female",
-    "2": "male"
+    "0": "adult female",
+    "1": "adult male",
+    "2": "child"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "child": "0",
-    "female": "1",
-    "male": "2"
+    "adult female": "0",
+    "adult male": "1",
+    "child": "2"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -30,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.41.2"
+  "transformers_version": "4.30.2"
 }

preprocessor_config.json

@@ -1,18 +1,4 @@
 {
-  "_valid_processor_keys": [
-    "images",
-    "do_resize",
-    "size",
-    "resample",
-    "do_rescale",
-    "rescale_factor",
-    "do_normalize",
-    "image_mean",
-    "image_std",
-    "return_tensors",
-    "data_format",
-    "input_data_format"
-  ],
   "do_normalize": true,
   "do_rescale": true,
   "do_resize": true,

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56f71c8a0b726235114f1e6ec07b78285f8daf93cca1eb5831d1b74449e873a6
+size 343269037