commit files to HF hub

README.md

@@ -21,32 +21,4 @@ model-index:
 # kitchen-layouts-2.2.0
 
 
-Kitchen Layouts detection🤗🖼️
-
-
-## Example Images
-
-
-#### g_shaped_kitchen
-
-![g_shaped_kitchen](images/g_shaped_kitchen.jpeg)
-
-#### galley_kitchen
-
-![galley_kitchen](images/galley_kitchen.jpeg)
-
-#### island_kitchen
-
-![island_kitchen](images/island_kitchen.jpeg)
-
-#### l_shaped_kitchen
-
-![l_shaped_kitchen](images/l_shaped_kitchen.jpeg)
-
-#### single_wall_kitchen
-
-![single_wall_kitchen](images/single_wall_kitchen.jpeg)
-
-#### u_shaped_kitchen
-
-![u_shaped_kitchen](images/u_shaped_kitchen.jpeg)
+Kitchen Layouts detection🤗🖼️

config.json

@@ -20,12 +20,12 @@
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "g_shaped_kitchen": "0",
-    "galley_kitchen": "1",
-    "island_kitchen": "2",
-    "l_shaped_kitchen": "3",
-    "single_wall_kitchen": "4",
-    "u_shaped_kitchen": "5"
+    "g_shaped_kitchen": 0,
+    "galley_kitchen": 1,
+    "island_kitchen": 2,
+    "l_shaped_kitchen": 3,
+    "single_wall_kitchen": 4,
+    "u_shaped_kitchen": 5
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -36,5 +36,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.38.2"
+  "transformers_version": "4.39.0"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0c5c666eabac3de6028a6a1df4d4c641bd5fd97b743b758212110468e7de5151
+oid sha256:b99a3e854df44e58bdd71865e8f9e87775dd2d992404f4162c2ffecdd165b030
 size 343236280

preprocessor_config.json

@@ -1,4 +1,18 @@
 {
+  "_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,
@@ -7,7 +21,7 @@
     0.5,
     0.5
   ],
-  "image_processor_type": "ViTFeatureExtractor",
+  "image_processor_type": "ViTImageProcessor",
   "image_std": [
     0.5,
     0.5,