Update README.md

README.md

@@ -14,9 +14,9 @@ tags:
 
 ![1](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/3QSs87msGNfDnPiiHsU1m.png)
 
-# **SigLIP2-ImageShield-90M**
+# **SigLIP2-ImageShield-90M-224**
 
-> **SigLIP2-ImageShield-90M** is a vision-language encoder model fine-tuned from **google/siglip2-base-patch16-224** for **multi-class image classification**. Built on the **SiglipForImageClassification** architecture, the model is designed to identify and categorize visual content for explicit, suggestive, and safe media filtering.
+> **SigLIP2-ImageShield-90M-224** is a vision-language encoder model fine-tuned from **google/siglip2-base-patch16-224** for **multi-class image classification**. Built on the **SiglipForImageClassification** architecture, the model is designed to identify and categorize visual content for explicit, suggestive, and safe media filtering.
 
 > [!note]
 > *SigLIP 2: Multilingual Vision-Language Encoders with Improved Semantic Understanding, Localization, and Dense Features*
@@ -49,7 +49,7 @@ from PIL import Image
 import torch
 
 # Load model and processor
-model_name = "prithivMLmods/SigLIP2-ImageShield-90M"  # Replace with your model path if needed
+model_name = "prithivMLmods/SigLIP2-ImageShield-90M-224"  # Replace with your model path if needed
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)