commit files to HF hub

.gitattributes

@@ -47,3 +47,13 @@ images/hotel_pool.jpg filter=lfs diff=lfs merge=lfs -text
 images/hotel_restaurants.jpg filter=lfs diff=lfs merge=lfs -text
 images/hotel_room.jpg filter=lfs diff=lfs merge=lfs -text
 images/hotel_suite.jpg filter=lfs diff=lfs merge=lfs -text
+images/bathroom.jpeg filter=lfs diff=lfs merge=lfs -text
+images/beach.jpg filter=lfs diff=lfs merge=lfs -text
+images/fitness.jpg filter=lfs diff=lfs merge=lfs -text
+images/food.jpg filter=lfs diff=lfs merge=lfs -text
+images/lobby.jpg filter=lfs diff=lfs merge=lfs -text
+images/meetings.jpg filter=lfs diff=lfs merge=lfs -text
+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

README.md

@@ -15,7 +15,7 @@ model-index:
     metrics:
       - name: Accuracy
         type: accuracy
-        value: 0.7105262875556946
+        value: 0.8064516186714172
 ---
 
 # hotel-image-classifier
@@ -27,38 +27,42 @@ Hotel Cateogry Transformer..
 ## Example Images
 
 
-#### hotel beach
+#### bathroom
 
-![hotel beach](images/hotel_beach.jpg)
+![bathroom](images/bathroom.jpeg)
 
-#### hotel fitness
+#### beach
 
-![hotel fitness](images/hotel_fitness.jpg)
+![beach](images/beach.jpg)
 
-#### hotel food
+#### fitness
 
-![hotel food](images/hotel_food.jpg)
+![fitness](images/fitness.jpg)
 
-#### hotel lobby
+#### food
 
-![hotel lobby](images/hotel_lobby.jpg)
+![food](images/food.jpg)
 
-#### hotel meetings
+#### lobby
 
-![hotel meetings](images/hotel_meetings.jpg)
+![lobby](images/lobby.jpg)
 
-#### hotel pool
+#### meetings
 
-![hotel pool](images/hotel_pool.jpg)
+![meetings](images/meetings.jpg)
 
-#### hotel restaurants
+#### pool
 
-![hotel restaurants](images/hotel_restaurants.jpg)
+![pool](images/pool.jpg)
 
-#### hotel room
+#### restaurants
 
-![hotel room](images/hotel_room.jpg)
+![restaurants](images/restaurants.jpg)
 
-#### hotel suite
+#### room
 
-![hotel suite](images/hotel_suite.jpg)
+![room](images/room.jpg)
+
+#### suite
+
+![suite](images/suite.jpg)

config.json

@@ -9,29 +9,31 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "hotel beach",
-    "1": "hotel fitness",
-    "2": "hotel food",
-    "3": "hotel lobby",
-    "4": "hotel meetings",
-    "5": "hotel pool",
-    "6": "hotel restaurants",
-    "7": "hotel room",
-    "8": "hotel suite"
+    "0": "bathroom",
+    "1": "beach",
+    "2": "fitness",
+    "3": "food",
+    "4": "lobby",
+    "5": "meetings",
+    "6": "pool",
+    "7": "restaurants",
+    "8": "room",
+    "9": "suite"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "hotel beach": "0",
-    "hotel fitness": "1",
-    "hotel food": "2",
-    "hotel lobby": "3",
-    "hotel meetings": "4",
-    "hotel pool": "5",
-    "hotel restaurants": "6",
-    "hotel room": "7",
-    "hotel suite": "8"
+    "bathroom": "0",
+    "beach": "1",
+    "fitness": "2",
+    "food": "3",
+    "lobby": "4",
+    "meetings": "5",
+    "pool": "6",
+    "restaurants": "7",
+    "room": "8",
+    "suite": "9"
   },
   "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:4cc5aded2d55b861b0766197426dbc7a28516841b9b3ebead739c6e18d7c13c7
-size 343287469
+oid sha256:5fea2c574cd88858f62435f3fd736cdc011c75b4602034067fdcc4c12024ade4
+size 343290541

runs/events.out.tfevents.1672101495.c5a8d66aacb4

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