commit files to HF hub

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 images/Weeping_Willow.jpg filter=lfs diff=lfs merge=lfs -text
+images/Pepper_Tree.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -31,6 +31,10 @@ Report any issues with the demo at the [github repo](https://github.com/nateraw/
 ## Example Images
 
 
+#### Pepper Tree
+
+![Pepper Tree](images/Pepper_Tree.jpg)
+
 #### Weeping Willow
 
 ![Weeping Willow](images/Weeping_Willow.jpg)

config.json

@@ -9,13 +9,15 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "Weeping Willow"
+    "0": "Pepper Tree",
+    "1": "Weeping Willow"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Weeping Willow": "0"
+    "Pepper Tree": "0",
+    "Weeping Willow": "1"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -23,7 +25,7 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
-  "problem_type": "regression",
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.30.2"

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e385f9e1ae756ccb95d2818e55cb1ef1decf16df9e7a84c1ffa5e8b5abcf6b81
-size 343262893
+oid sha256:509b03c906a8763e1e96dcc7c26aed536f474be34c3187a6b27e75cbb980d13a
+size 343265965

runs/events.out.tfevents.1686989556.669add52608c.207.15

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46edabfa2ddf949a77db6acfe3ef512b5296ad0d8a6f53b6150a607c4d03b523
+size 929