Update README.md

README.md

@@ -15,13 +15,6 @@ widget:
   example_title: Savanna
 ---
 
-# CO2
-
-This model was trained on an M1 and took 0.322 g of CO2 (measured with [CodeCarbon](https://codecarbon.io/))
-
-
-# Results
-
 This model is a fine-tuned version of [facebook/deit-tiny-distilled-patch16-224](https://huggingface.co/facebook/deit-tiny-distilled-patch16-224) on the [docornot](https://huggingface.co/datasets/tarekziade/docornot) dataset.
 
 It achieves the following results on the evaluation set:
@@ -29,22 +22,24 @@ It achieves the following results on the evaluation set:
 - Accuracy: 1.0
 
 
+# CO2 emissions
 
+This model was trained on an M1 and took 0.322 g of CO2 (measured with [CodeCarbon](https://codecarbon.io/))
 
-## Model description
+# Model description
 
 This model is distilled Vision Transformer (ViT) model. 
 Images are presented to the model as a sequence of fixed-size patches (resolution 16x16), which are linearly embedded. 
 
-## Intended uses & limitations
+# Intended uses & limitations
 
 You can use this model to detect if an image is a picture or a document.
 
-## Training procedure
+# Training procedure
 
 Source code used to generate this model : https://github.com/tarekziade/docornot
 
-### Training hyperparameters
+## Training hyperparameters
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
@@ -55,14 +50,14 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 1
 
-### Training results
+## Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
 | 0.0           | 1.0   | 1600 | 0.0000          | 1.0      |
 
 
-### Framework versions
+## Framework versions
 
 - Transformers 4.39.2
 - Pytorch 2.2.2