Update README.md

README.md

@@ -1,7 +1,7 @@
 ---
 language: en
 datasets:
-- custom
+- abdulmananraja/real-life-violence-situations
 tags:
 - image-classification
 - vision
@@ -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` and has been trained on a custom dataset 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