Update README.md

README.md

@@ -7,32 +7,40 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
-model-index:
-- name: dinov2-small-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. -->
 
-# dinov2-small-types-of-film-shots-vN
+# dinov2-small: types of film shots
+
+
+## Model description
 
 This model is a fine-tuned version of [facebook/dinov2-small](https://huggingface.co/facebook/dinov2-small) on the szymonrucinski/types-of-film-shots dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.9864
 - Accuracy: 0.6259
 
-## 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
 
@@ -74,4 +82,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