Update README.md

README.md

@@ -6,6 +6,9 @@ datasets:
 - imagefolder
 metrics:
 - accuracy
+- f1
+- recall
+- precision
 model-index:
 - name: vit-base-patch16-224-in21k_vegetables_clf
   results:
@@ -21,15 +24,15 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 1.0
+      value: 1
+language:
+- en
+pipeline_tag: image-classification
 ---
 
-<!-- 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. -->
-
 # vit-base-patch16-224-in21k_vegetables_clf
 
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k).
 It achieves the following results on the evaluation set:
 - Loss: 0.0014
 - Accuracy: 1.0
@@ -45,15 +48,17 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+This is a multiclass image classification model of different vegetables.
+
+For more information on how it was created, check out the following link: https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Vegetable%20Image%20Classification/Vegetables_ViT.ipynb
 
 ## Intended uses & limitations
 
-More information needed
+This model is intended to demonstrate my ability to solve a complex problem using technology.
 
 ## Training and evaluation data
 
-More information needed
+Dataset Source: https://www.kaggle.com/datasets/misrakahmed/vegetable-image-dataset
 
 ## Training procedure
 
@@ -82,4 +87,4 @@ The following hyperparameters were used during training:
 - Transformers 4.25.1
 - Pytorch 1.12.1
 - Datasets 2.8.0
-- Tokenizers 0.12.1
+- Tokenizers 0.12.1