Update README.md

README.md

@@ -4,35 +4,44 @@ base_model: microsoft/beit-large-patch16-512
 tags:
 - image-classification
 - vision
-- generated_from_trainer
+- cinematography
 metrics:
 - accuracy
-model-index:
-- name: beit-large-patch16-512-types-of-film-shots-vN
-  results: []
+datasets:
+- szymonrucinski/types-of-film-shots
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# beit-large-patch16-512-types-of-film-shots-vN
+# beit-large-patch16-512: types-of-film-shots
+
+
+## Model description
+
 
 This model is a fine-tuned version of [microsoft/beit-large-patch16-512](https://huggingface.co/microsoft/beit-large-patch16-512) on the szymonrucinski/types-of-film-shots dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.2335
 - Accuracy: 0.6763
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
 
-More information needed
+## class labels
 
-## Training and evaluation data
+The dataset contains the following labels:
 
-More information needed
+```json
+"id2label": {
+    "0": "ambiguous",
+    "1": "closeUp",
+    "2": "detail",
+    "3": "extremeLongShot",
+    "4": "fullShot",
+    "5": "longShot",
+    "6": "mediumCloseUp",
+    "7": "mediumShot"
+  },
+```
 
 ## Training procedure
 
@@ -66,4 +75,4 @@ The following hyperparameters were used during training:
 - Transformers 4.38.0.dev0
 - Pytorch 2.2.0+cu121
 - Datasets 2.17.1
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2