Update README.md

README.md

@@ -38,11 +38,11 @@ from transformers import ViTForImageClassification, ViTFeatureExtractor
 from PIL import Image
 
 # Load the model and feature extractor
-model = ViTForImageClassification.from_pretrained('your-username/vit-base-violence')
-feature_extractor = ViTFeatureExtractor.from_pretrained('your-username/vit-base-violence')
+model = ViTForImageClassification.from_pretrained('jaranohaal/vit-base-violence-detection')
+feature_extractor = ViTFeatureExtractor.from_pretrained('jaranohaal/vit-base-violence-detection')
 
 # Load an image
-image = Image.open('path_to_your_image.jpg')
+image = Image.open('image.jpg')
 
 # Preprocess the image
 inputs = feature_extractor(images=image, return_tensors="pt")