Update README.md

README.md

@@ -1,21 +1,25 @@
 ---
 language: en
 tags:
-  - vision
-  - image-classification
-  - pokemon
-  - pytorch
-  - transformers
+- vision
+- image-classification
+- pokemon
+- pytorch
+- transformers
 license: apache-2.0
 datasets:
-  - custom
+- custom
 pipeline_tag: image-classification
-model_name: "Pokemon Classifier Gen9"
-model_id: "skshmjn/Pokemon-classifier-gen9-1025"
-library_name: "transformers"
-framework: "PyTorch"
+model_name: Pokemon Classifier Gen9
+model_id: skshmjn/Pokemon-classifier-gen9-1025
+library_name: transformers
+framework: PyTorch
 widget:
-  - src: "test.jpg"
+- src: test.jpg
+metrics:
+- accuracy
+base_model:
+- google/vit-base-patch16-224-in21k
 ---
 
 # Model Card for Pokemon Classifier Gen9
@@ -59,4 +63,4 @@ predicted_pokemon = model.config.id2label[predicted_id]
 
 # Print predicted class
 print(f"Predicted Pokémon Pokédex number: {predicted_id+1}")
-print(f"Predicted Pokémon: {predicted_pokemon}")
+print(f"Predicted Pokémon: {predicted_pokemon}")