Update README.md

README.md

@@ -1,3 +1,9 @@
+---
+license: mit
+base_model:
+- google/vit-large-patch16-224
+pipeline_tag: image-classification
+---
 # Pokemon Team Classification with Vision Transformer
 
 A fine-tuned Vision Transformer (ViT) model for classifying 6 specific Pokemon from a competitive team setup. This model can identify Arceus, Marshadow, Sandy Shocks, Slaking, Reshiram, and Magearna with high accuracy.
@@ -138,12 +144,4 @@ This model is released under the MIT License. The training data consists of Poke
 
 ---
 
-**Note**: This model is specifically trained for a competitive Pokemon team setup and may not generalize to other Pokemon or use cases. For broader Pokemon classification, consider training on a more comprehensive dataset.
-
-
----
-license: mit
-base_model:
-- google/vit-large-patch16-224
-pipeline_tag: image-classification
----
+**Note**: This model is specifically trained for a competitive Pokemon team setup and may not generalize to other Pokemon or use cases. For broader Pokemon classification, consider training on a more comprehensive dataset.