commit files to HF hub

README.md

@@ -3,9 +3,11 @@ tags:
 - image-classification
 - pytorch
 - huggingpics
+metrics:
+- accuracy
 
-model-index:  
-- name: baked-goods-2
+model-index:
+- name: baked-goods
   results:
   - task:
       name: Image Classification
@@ -13,9 +15,24 @@ model-index:
     metrics:
       - name: Accuracy
         type: accuracy
-        value: 0.8035714030265808
+        value: 0.875
 ---
 
-# baked-goods-2
+# baked-goods
 
-Autogenerated by a super cool Colab notebook 😎
+Autogenerated by HuggingPics🤗🖼️
+
+## Example Images
+
+
+#### cake
+
+![cake](images/cake.jpg)
+
+#### cookie
+
+![cookie](images/cookie.jpg)
+
+#### pie
+
+![pie](images/pie.jpg)

config.json

@@ -8,21 +8,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "brownie",
-    "1": "cake",
-    "2": "cookie",
-    "3": "donut",
-    "4": "pie"
+    "0": "cake",
+    "1": "cookie",
+    "2": "pie"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "brownie": "0",
-    "cake": "1",
-    "cookie": "2",
-    "donut": "3",
-    "pie": "4"
+    "cake": "0",
+    "cookie": "1",
+    "pie": "2"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -30,5 +26,5 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
-  "transformers_version": "4.8.0"
+  "transformers_version": "4.8.1"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cac94f85512c3030c43d13f4971769e68166f0b78bd61c23fe2d6ddde1a496f0
-size 343286257
+oid sha256:fb3714a0379c159424e2fb64ecb0dc7f72bc7c062bfc4156d2e6e34e285c11b7
+size 343280113