Update README.md

README.md

@@ -61,7 +61,7 @@ inference:
 ---
 # vision-1-mini
 
-Vision-1-mini is an optimized 8B parameter model based on Llama 3.1, specifically designed for brand safety classification. This model is particularly optimized for Apple Silicon devices and provides efficient, accurate brand safety assessments using the BrandSafe-16k classification system.
+Vision-1-mini is an optimized 8B parameter model based on Llama 3.1, specifically designed for brand safety classification, trained on our [BrandSafe-16k](https://huggingface.co/datasets/OverseerAI/BrandSafe-16k) dataset. This model is particularly optimized for Apple Silicon devices and provides efficient, accurate brand safety assessments using the BrandSafe-16k classification system.
 
 ## Model Details