commit files to HF hub

.gitattributes

@@ -57,3 +57,5 @@ images/pool.jpg filter=lfs diff=lfs merge=lfs -text
 images/restaurants.jpg filter=lfs diff=lfs merge=lfs -text
 images/room.jpg filter=lfs diff=lfs merge=lfs -text
 images/suite.jpg filter=lfs diff=lfs merge=lfs -text
+images/bathroom.jpg filter=lfs diff=lfs merge=lfs -text
+images/spa.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -15,7 +15,7 @@ model-index:
     metrics:
       - name: Accuracy
         type: accuracy
-        value: 0.8064516186714172
+        value: 0.8020833134651184
 ---
 
 # hotel-image-classifier
@@ -29,7 +29,7 @@ Hotel Cateogry Transformer..
 
 #### bathroom
 
-![bathroom](images/bathroom.jpeg)
+![bathroom](images/bathroom.jpg)
 
 #### beach
 
@@ -63,6 +63,10 @@ Hotel Cateogry Transformer..
 
 ![room](images/room.jpg)
 
+#### spa
+
+![spa](images/spa.jpg)
+
 #### suite
 
 ![suite](images/suite.jpg)

config.json

@@ -11,6 +11,7 @@
   "id2label": {
     "0": "bathroom",
     "1": "beach",
+    "10": "suite",
     "2": "fitness",
     "3": "food",
     "4": "lobby",
@@ -18,7 +19,7 @@
     "6": "pool",
     "7": "restaurants",
     "8": "room",
-    "9": "suite"
+    "9": "spa"
   },
   "image_size": 224,
   "initializer_range": 0.02,
@@ -33,7 +34,8 @@
     "pool": "6",
     "restaurants": "7",
     "room": "8",
-    "suite": "9"
+    "spa": "9",
+    "suite": "10"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5fea2c574cd88858f62435f3fd736cdc011c75b4602034067fdcc4c12024ade4
-size 343290541
+oid sha256:8611aa3dd696862e5daf8ef52ca20f780fd61aa3df40885e1d3ce91d710a2233
+size 343293613

runs/events.out.tfevents.1672105026.c5a8d66aacb4

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