Upload 3 files

config.json

@@ -1,5 +1,6 @@
 {
-  "_name_or_path": "./out/model/final_model",
+  "_name_or_path": "0-ma/mobilenet-v2-geometric-shapes",
+  "_num_labels": 6,
   "architectures": [
     "MobileNetV2ForImageClassification"
   ],
@@ -21,12 +22,12 @@
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "1": "0",
-    "2": "1",
-    "3": "2",
-    "4": "3",
-    "5": "4",
-    "6": "5"
+    "1": 0,
+    "2": 1,
+    "3": 2,
+    "4": 3,
+    "5": 4,
+    "6": 5
   },
   "layer_norm_eps": 0.001,
   "min_depth": 8,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:49c0582b6f59c0cac79962a04bb630bc84a3c0cb727daddc9b6ef3f65db7530d
+oid sha256:d59ceae7a99ffab3030d187f3e02ac39119ad086f6786b05f0ec28500fe85761
 size 9100712

preprocessor_config.json

@@ -23,4 +23,4 @@
   "size": {
     "shortest_edge": 256
   }
-}
+}