Update README.md

README.md

@@ -8,14 +8,14 @@ tags:
 metrics:
 - accuracy
 model-index:
-- name: Wildfire Dectection
+- name: wildfire-classifier
   results: []
 ---
 
 <!-- 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. -->
 
-# fine_tuned_wildfire
+# Wildfire classifier
 
 This model is a fine-tuned version of [google/vit-base-patch16-384](https://huggingface.co/google/vit-base-patch16-384) on the 
 [Kaggle Wildfire Dataset](https://www.kaggle.com/datasets/elmadafri/the-wildfire-dataset) dataset.
@@ -23,19 +23,8 @@ It achieves the following results on the evaluation set:
 - Loss: 0.2329
 - Accuracy: 0.9202
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
+## Fine-tunning
+TOOD: Add finetuning notebook
 
 ### Training hyperparameters