Upload 2 files

config.json

@@ -10,26 +10,24 @@
   "hidden_size": 768,
   "id2label": {
     "0": "Round",
-    "1": "Collared",
-    "2": "High neck",
-    "3": "V Neck",
-    "4": "Straight",
+    "1": "VNeck",
+    "2": "Straight",
+    "3": "Hoodie",
+    "4": "Henley",
     "5": "Halter",
-    "6": "Asymmetric",
-    "7": "Henley"
+    "6": "Sweetheart"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Asymmetric": 6,
-    "Collared": 1,
     "Halter": 5,
-    "Henley": 7,
-    "High neck": 2,
+    "Henley": 4,
+    "Hoodie": 3,
     "Round": 0,
-    "Straight": 4,
-    "V Neck": 3
+    "Straight": 2,
+    "Sweetheart": 6,
+    "VNeck": 1
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -37,7 +35,6 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
-  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.44.2"