Update README.md

README.md

@@ -1,13 +1,19 @@
 ---
 license: apache-2.0
 tags:
+- image-classification
 - generated_from_trainer
 datasets:
 - beans
 metrics:
 - accuracy
+widget:
+- src: https://huggingface.co/platzi/platzi-vit-model-andres-galvis/blob/main/healthy.jpeg
+  example_title: Healthy
+- src: https://huggingface.co/platzi/platzi-vit-model-andres-galvis/blob/main/bean_rust.jpeg
+  example_title: Bean Rust
 model-index:
-- name: platzi-vit-model-andres-galvis
+- name: platzi-vit-base-beans
   results:
   - task:
       name: Image Classification
@@ -16,7 +22,7 @@ model-index:
       name: beans
       type: beans
       config: default
-      split: validation
+      split: train
       args: default
     metrics:
     - name: Accuracy