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/Narendra_Modi.png filter=lfs diff=lfs merge=lfs -text
+images/Lokesh.png 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
 
 
+#### Lokesh
+
+![Lokesh](images/Lokesh.png)
+
 #### Narendra_Modi
 
 ![Narendra_Modi](images/Narendra_Modi.png)

config.json

@@ -9,13 +9,15 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "Narendra_Modi"
+    "0": "Lokesh",
+    "1": "Narendra_Modi"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Narendra_Modi": "0"
+    "Lokesh": "0",
+    "Narendra_Modi": "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.35.2"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f267420c825af45bad3fbace50b0a1f989fb40a1a95d3ca678f6f3441bfe82fb
-size 343220892
+oid sha256:8dcaf2981ce5e4a9a18c1d3f763d55e6b2c545cdba978c5f7680732ec0e9af33
+size 343223968

runs/events.out.tfevents.1707127200.50ba1c9dfc9d.1058.8

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