Update README.md

README.md

@@ -13,7 +13,7 @@ license: apache-2.0
 
 ## Model Description
 
-This is a Vision Transformer (ViT) model fine-tuned for violence detection. The model is based on `google/vit-base-patch16-224-in21k`[https://huggingface.co/google/vit-base-patch16-224-in21k] and has been trained on the "Real Life Violence Situations"[https://www.kaggle.com/datasets/mohamedmustafa/real-life-violence-situations-dataset] dataset from Kaggle to classify images into violent or non-violent categories.
+This is a Vision Transformer (ViT) model fine-tuned for violence detection. The model is based on [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) and has been trained on the [Real Life Violence Situations](https://www.kaggle.com/datasets/mohamedmustafa/real-life-violence-situations-dataset) dataset from Kaggle to classify images into violent or non-violent categories.
 
 ## Intended Use
 
@@ -22,6 +22,11 @@ The model is intended for use in applications where detecting violent content in
 - Content moderation
 - Surveillance
 - Parental control software
+  
+## Model accuracy
+
+Test accuracy for Vit Base = 98.80%
+Loss = 0.20038144290447235
 
 ## How to Use