Update README.md

README.md

@@ -1,3 +1,9 @@
+---
+license: mit
+datasets: AllenTAN/image_sentiment
+base_model: google/efficientnet-b2
+---
+
 # EfficientNet B2 Image Classification
 
 This project implements an image classification model using the EfficientNet B2 architecture, fine-tuned on a custom dataset. It provides a modular and easy-to-use structure for training and evaluating the model.
@@ -73,10 +79,4 @@ Feel free to open issues or submit pull requests if you have suggestions for imp
 
 ## License
 
-MIT License
-
----
-license: "mit"
-datasets: "AllenTAN/image_sentiment"
-base_model: "google/efficientnet-b2"
----
+MIT License